search

VITA-Group / EnlightenGAN

[IEEE TIP] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang
Other
890 stars 198 forks source link

Pretrained Discriminator for Continued Training #48

Closed surajpaib closed 4 years ago

surajpaib commented 4 years ago

Thank you for sharing this great project! I've been working for the past week on experimenting with your codebase and trying out object-specific enhancement on the Exclusively Dark Dataset. I do have some decent results but they do not look qualitatively the same as the ones from the Arxiv paper for the domain adapted BBD dataset.

I'd like to try continued training on the EnlightenGAN but can't seem to find the Discriminator weights. Would you mind pointing me in the right direction or sharing these weights if they aren't public?

yifanjiang19 commented 4 years ago

Actually I can not find the discriminators weights now. Normally if you set the dark dataset using your data and set bright dataset using the dataset I provided, you should be able to reproduce the results following the README guideline.

surajpaib commented 4 years ago

@yueruchen No problem! Just wanted to see if these files exist somewhere so that I can easily plug and play to continue training. Thanks for checking.

Closing the issue now