Models

English | 简体中文

Contents

1. Supported Models
2. Inheritance Relationship

Supported Models

Class	Description	Supported Algorithms
BaseModel	Abstract base class, define common functions
SRModel	Base image SR class	SRCNN, EDSR, SRResNet, RCAN, RRDBNet, etc
SRGANModel	SRGAN image SR class	SRGAN
ESRGANModel	ESRGAN image SR class	ESRGAN
VideoBaseModel	Base video SR class
EDVRModel	EDVR video SR class	EDVR
StyleGAN2Model	StyleGAN2 generation class	StyleGAN2

Inheritance Relationship

In order to reuse components among models, we use a lot of inheritance. The following is the inheritance relationship:

BaseModel
├── SRModel
│   ├── SRGANModel
│   │   └── ESRGANModel
│   └── VideoBaseModel
│       ├── VideoGANModel
│       └── EDVRModel
└── StyleGAN2Model