Skip to content

Latest commit

 

History

History
35 lines (27 loc) · 1.28 KB

Models.md

File metadata and controls

35 lines (27 loc) · 1.28 KB

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