Add BasicVSR (#245)

* Add BasicVSR

* fix isort

* fix isort

* rename to basicvsr_net.py

* change step_counter to a key in train_cfg

* rename and fix minor bug

* rename to BasicVSRNet

* Fix init and model

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'
 ]

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'
+]