search

swz30 / MIRNet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.
Other
667 stars 96 forks source link

Codes about iamge enhancement #11

Closed xuhui1994 closed 3 years ago

xuhui1994 commented 3 years ago

Thank you for your excellent work about image filed firstly? In this repo， only train_denosing.py can be seen . Could you provide image enhancement training .py code? Looking forward to your reply.

swz30 commented 3 years ago

Train_denoising.py file is provided as an example. You can use the same training code for image enhancement as well. Just provide the paths to your training and validation data here https://github.com/swz30/MIRNet/blob/d7e471704fcdb67526af027352c1a5ea4619ee00/training.yml#L23

swz30 commented 3 years ago

Please set NUM_EPOCHS to 200 for Image Enhancement training.