Skip to content

AOD-NET Implementation on the STM32H7A3ZI for On-Edge Single Image Dehazing

License

Notifications You must be signed in to change notification settings

adnan-armouti/AOD_Net_on_Edge

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AOD-NET Implementation on STM32H7A3ZI Device with ARM Cortex M7 for On-Edge Single Image Dehazing

Contributing Author: Adnan Armouti.

Pre-requisite Hardware and Software

This section is pertinent to those who wish to run this code as developed. All following instructions are with respect to the hardware used by the author.

Prior to running the code, please ensure the following:

(1) STM32H7A3ZI Board

Please purchase this device link. You will also need this accessory link. This will be used to connect the H7 board to a FAT16/FAT32 formatted USB flash drive. For more information, please refer to this link: link.

(2) STM32 CUBE IDE

Please download the STM32 CUBE IDE from this link. Online tutorials on navigating this development environment can be found extensively on Youtube. Code is developed in C.


Input Data

This section refers to where the model parameters and input image is currently stored. This data must be transferred to the USB flash drive, and connected to the H7 board.

(1) Model Parameters

The folder "params_utils" contains the model parameter files. Weights files are denoted by "w", and bias files are denoted by "b". These files must be saved on the USB stick for the program to read them in. These model parameters were trained using PyTorch, and saved for the forward pass implementation in this project.

(2) Input Hazy Image

The folder "params_utils" also contains some example hazy images that can be used/exchanged/removed. Some dehazed examples are also provided. Please note that given the memory constraints of the device, only low-resolution images are accepted as input.


Execution

The folder "cnn_stm32h743" contains the H7 files (and .ioc file for STM CUBE IDE). Please compile and run the cnn_stm32h743/NN_miniproject_demo/Core/Src/main.c file.


Citations

  1. Li, Boyi, et al. "Aod-net: All-in-one dehazing network." Proceedings of the IEEE international conference on computer vision. 2017.

About

AOD-NET Implementation on the STM32H7A3ZI for On-Edge Single Image Dehazing

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages