Giters

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About self-regularized attention map！

wanlei19961229 opened this issue · comments

commented

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

commented

Here it is

EnlightenGAN/models/networks.py

Line 756 in 90903c3

x = x*gray_5 if self.opt.self_attention else x