search

Li-Chongyi / Zero-DCE

Zero-DCE code and model
774 stars 187 forks source link

Training on custom data and outputs #59

Open rameshshreyas opened 10 months ago

rameshshreyas commented 10 months ago

Hi @Li-Chongyi,

Thanks for this implementation! I have also implemented your paper and trained on my data that consists of images of road at night. The output from ZeroDCE seems to blow up some of the brighter ones as it tries to brighten the darker patches. I used your pretrained model as well as trained on a combination of data from LoL, as well as my vehicle datasets.

2023-10-17-06-16-22_002675_enhanced 2023-10-17-06-16-22_002675_original

Any pointers to tweak the loss weights would be greatly appreciated.