Update NAFNet_arch.py

2022-04-12 11:32:49 +08:00 · 2022-04-12 11:32:49 +08:00 · 8635f895da
parent f2c97154b0
commit 8635f895da
1 changed files with 6 additions and 0 deletions
--- a/basicsr/models/archs/NAFNet_arch.py
+++ b/basicsr/models/archs/NAFNet_arch.py
@ -5,6 +5,12 @@
 '''
 Simple Baselines for Image Restoration

+@article{chen2022simple,
+  title={Simple Baselines for Image Restoration},
+  author={Chen, Liangyu and Chu, Xiaojie and Zhang, Xiangyu and Sun, Jian},
+  journal={arXiv preprint arXiv:2204.04676},
+  year={2022}
+}
 '''

 import torch