CSDNet-CSDGAN

this is the code for the paper "Learning Deep Context-Sensitive Decomposition for Low-Light Image Enhancement"

Environment Preparing

python 3.6
pytorch 0.4.1

Testing

Download pretrained model from Google drive or Baidu drive (extraction code:xcch). Then put them into ./checkpoints/

Finally, run the script below, the results will be saved in ./results/

python test.py 
--dataroot           #The folder path of the picture you want to test
E:/test/
--name               #The checkpoint name
CSDNet_UPE or CSDNet_LOL or CSDGAN or LiteCSDNet_UPE or LiteCSDNet_LOL or SLiteCSDNet_UPE or SLiteCSDNet_LOL
--gpu_ids            #could be single or multiple
0

A great thanks to EnlightenGAN for providing the basis for this code.