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

About self-regularized attention map! #91

Closed wanlei19961229 closed 3 years ago

wanlei19961229 commented 3 years ago

Hi,can you tell me where does the self-regularized attention map generate in the code? I cannot find it.Thank you very much.

yifanjiang19 commented 3 years ago

Here it is https://github.com/VITA-Group/EnlightenGAN/blob/90903c3f9086583baf450474505d4dd1d5495ae0/models/networks.py#L756