From a74fb5751cd002481d78cec4d3a118fc47d42605 Mon Sep 17 00:00:00 2001 From: RebornForPower Date: Tue, 9 Nov 2021 19:53:43 +0800 Subject: [PATCH] Add reference --- basicsr/archs/focalir_arch.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/basicsr/archs/focalir_arch.py b/basicsr/archs/focalir_arch.py index 33f92ee45..1c2aa6886 100644 --- a/basicsr/archs/focalir_arch.py +++ b/basicsr/archs/focalir_arch.py @@ -1,3 +1,8 @@ +###### +# FocalIR +# This code is referenced by Focal Transformer and SwinIR +# This model is supported by BasicSR +###### # -------------------------------------------------------- # Focal Transformer # Copyright (c) 2021 Microsoft @@ -5,6 +10,11 @@ # Written by Jianwei Yang (jianwyan@microsoft.com) # Based on Swin Transformer written by Zhe Liu # -------------------------------------------------------- +# ----------------------------------------------------------------------------------- +# SwinIR: Image Restoration Using Swin Transformer, https://arxiv.org/abs/2108.10257 +# Originally Written by Ze Liu, Modified by Jingyun Liang. +# ----------------------------------------------------------------------------------- + import math import torch