mmagic.models.editors.edvr.edvr
¶
Module Contents¶
Classes¶
EDVR model for video super-resolution. |
- class mmagic.models.editors.edvr.edvr.EDVR(generator, pixel_loss, train_cfg=None, test_cfg=None, init_cfg=None, data_preprocessor=None)[source]¶
Bases:
mmagic.models.BaseEditModel
EDVR model for video super-resolution.
EDVR: Video Restoration with Enhanced Deformable Convolutional Networks.
- Parameters
generator (dict) – Config for the generator structure.
pixel_loss (dict) – Config for pixel-wise loss.
train_cfg (dict) – Config for training. Default: None.
test_cfg (dict) – Config for testing. Default: None.
init_cfg (dict, optional) – The weight initialized config for
BaseModule
.data_preprocessor (dict, optional) – The pre-process config of
BaseDataPreprocessor
.
- forward_train(inputs, data_samples=None)[source]¶
Forward training. Returns dict of losses of training.
- Parameters
inputs (torch.Tensor) – batch input tensor collated by
data_preprocessor
.data_samples (List[BaseDataElement], optional) – data samples collated by
data_preprocessor
.
- Returns
Dict of losses.
- Return type
dict