类 | 任务 | 训练/测试 | 描述 |
---|---|---|---|
PairedImageDataset | 图像超分 | 训练 | 支持读取成对的训练数据 |
SingleImageDataset | 图像超分 | 测试 | 只读取low quality的图像, 用在没有Ground-Truth的测试中 |
REDSDataset | 视频超分 | 训练 | REDS的训练数据集 |
Vimeo90KDataset | 视频超分 | 训练 | Vimeo90K的训练数据集 |
VideoTestDataset | 视频超分 | 测试 | 基础的视频超分测试集, 支持Vid4, REDS测试集 |
VideoTestVimeo90KDataset | 视频超分 | 测试 | 继承VideoTestDataset , Vimeo90K的测试数据集 |
VideoTestDUFDataset | 视频超分 | 测试 | 继承VideoTestDataset , 方法DUF的测试数据集, 支持Vid4 |
FFHQDataset | 人脸生成 | 训练 | FFHQ的训练数据集 |
- 共用的变换和函数分别在 transforms.py 和 util.py 中