[Feature] Support FLAVR Net (open-mmlab#866)

* [Feature] Support FLAVR Net * Update * Update * Update * Update * Update
wangruohui · Jul 7, 2022 · 4597b56 · 4597b56
1 parent 172cca8
commit 4597b56
Show file tree

Hide file tree

Showing 4 changed files with 633 additions and 3 deletions.
diff --git a/mmedit/models/backbones/__init__.py b/mmedit/models/backbones/__init__.py
@@ -15,7 +15,7 @@
                            BasicVSRPlusPlus, DICNet, EDVRNet, GLEANStyleGANv2,
                            IconVSR, MSRResNet, RealBasicVSRNet, RRDBNet,
                            TDANNet, TOFlow, TTSRNet)
-from .vfi_backbones import CAINNet, TOFlowVFINet
+from .vfi_backbones import CAINNet, FLAVRNet, TOFlowVFINet
 
 __all__ = [
     'MSRResNet', 'VGG16', 'PlainDecoder', 'SimpleEncoderDecoder',
@@ -29,5 +29,5 @@
     'UnetGenerator', 'ResnetGenerator', 'FBAResnetDilated', 'FBADecoder',
     'BasicVSRNet', 'IconVSR', 'TTSRNet', 'GLEANStyleGANv2', 'TDANNet',
     'LIIFEDSR', 'LIIFRDN', 'BasicVSRPlusPlus', 'RealBasicVSRNet', 'CAINNet',
-    'TOFlowVFINet'
+    'TOFlowVFINet', 'FLAVRNet'
 ]
diff --git a/mmedit/models/backbones/vfi_backbones/__init__.py b/mmedit/models/backbones/vfi_backbones/__init__.py
@@ -1,5 +1,6 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from .cain_net import CAINNet
+from .flavr_net import FLAVRNet
 from .tof_vfi_net import TOFlowVFINet
 
-__all__ = ['CAINNet', 'TOFlowVFINet']
+__all__ = ['CAINNet', 'TOFlowVFINet', 'FLAVRNet']