Merge f7e8050 into 9243ffc

open-mmlab · Apr 12, 2021 · 328867a · 328867a
2 parents 9243ffc + f7e8050
commit 328867a
Show file tree

Hide file tree

Showing 7 changed files with 794 additions and 4 deletions.
diff --git a/mmedit/models/backbones/__init__.py b/mmedit/models/backbones/__init__.py
@@ -12,7 +12,8 @@
                                SimpleEncoderDecoder)
 # yapf: enable
 from .generation_backbones import ResnetGenerator, UnetGenerator
-from .sr_backbones import EDSR, SRCNN, EDVRNet, MSRResNet, RRDBNet, TOFlow
+from .sr_backbones import (EDSR, SRCNN, BasicVSRNet, EDVRNet, MSRResNet,
+                           RRDBNet, TOFlow)
 
 __all__ = [
     'MSRResNet', 'VGG16', 'PlainDecoder', 'SimpleEncoderDecoder',
@@ -23,5 +24,6 @@
     'ContextualAttentionNeck', 'DeepFillDecoder', 'EDSR',
     'DeepFillEncoderDecoder', 'EDVRNet', 'IndexedUpsample', 'IndexNetEncoder',
     'IndexNetDecoder', 'TOFlow', 'ResGCAEncoder', 'ResGCADecoder', 'SRCNN',
-    'UnetGenerator', 'ResnetGenerator', 'FBAResnetDilated', 'FBADecoder'
+    'UnetGenerator', 'ResnetGenerator', 'FBAResnetDilated', 'FBADecoder',
+    'BasicVSRNet'
 ]
diff --git a/mmedit/models/backbones/sr_backbones/__init__.py b/mmedit/models/backbones/sr_backbones/__init__.py
@@ -1,8 +1,11 @@
+from .basicvsr_net import BasicVSRNet
 from .edsr import EDSR
 from .edvr_net import EDVRNet
 from .rrdb_net import RRDBNet
 from .sr_resnet import MSRResNet
 from .srcnn import SRCNN
 from .tof import TOFlow
 
-__all__ = ['MSRResNet', 'RRDBNet', 'EDSR', 'EDVRNet', 'TOFlow', 'SRCNN']
+__all__ = [
+    'MSRResNet', 'RRDBNet', 'EDSR', 'EDVRNet', 'TOFlow', 'SRCNN', 'BasicVSRNet'
+]