Zj-BinXia / ENLCA

This project is official implementation of 'Efficient Non-Local Contrastive Attention for Image Super-Resolution', AAAI2022
86 stars 7 forks source link

about ENLCA? #3

Closed fanrz closed 2 years ago

fanrz commented 2 years ago

thanks for your work. I have on question. in your paper, For each ordered sequence, we take the top n1N related ones as relevant features and n1N unrelated ones starting from n2N as irrelevant features. u use positive = torch.mean(score[:, :, :5], dim=2) negative = torch.mean(score[:, :, 50:], dim=2) # [N,H*W] why do you use 5 and 50?

Zj-BinXia commented 2 years ago

Sorry, we arrange the code after the paper being accepted. There may be the gap between the paper and code. You can change the code to "positive = torch.mean(score[:, :, :15], dim=2) negative = torch.mean(score[:, :, 50:65], dim=2) # [N,H*W] ".

fanrz commented 2 years ago

thank for your knid reply. I want to know for what reason, you pick 15 50 65. any explanation for the values? or observation of the training?

Zj-BinXia commented 2 years ago

observation of the training

fanrz commented 2 years ago

great, thanks