Skip to content

Latest commit

 

History

History
45 lines (23 loc) · 2.78 KB

File metadata and controls

45 lines (23 loc) · 2.78 KB

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.