Attention-based network for low-light image enhancement
This is a pytorch implementation of ALEN in ICME2020
This is the link of pre-trained model and test results in OneDrive and BaiduNetDisk(2fex).
- download the pre-trained model
- prepare test dataset
- run : python main.py --phase=test
requirement: python == 3.7 pytorch == 1.0.0 rawpy
You need change the dataset path when testing your images, and you need pay attention to the pack function in dataset.py when you test raw images captured by other devices.
prapare the dataset and train the network. That's all.
If you use our code, please cite our paper