| 类 | 描述 | 支持的算法 |
|---|---|---|
| BaseModel | 抽象基类, 定义了共有的函数 | |
| SRModel | 基础图像超分类 | SRCNN, EDSR, SRResNet, RCAN, RRDBNet, etc |
| SRGANModel | SRGAN图像超分类 | SRGAN |
| ESRGANModel | ESRGAN图像超分类 | ESRGAN |
| VideoBaseModel | 基础视频超分类 | |
| EDVRModel | EDVR视频超分类 | EDVR |
| StyleGAN2Model | StyleGAN2图像生成类 | StyleGAN2 |
为增加模型间的复用, 我们大量使用了继承, 以下为各个模型之间的继承关系:
BaseModel
├── SRModel
│ ├── SRGANModel
│ │ ├── ESRGANModel
│ ├── VideoBaseModel
│ ├── EDVRModel
├── StyleGAN2Model