在图像超分和视频超分领域中,数据集至关重要。本文旨在概述流行的开源图像超分和视频超分数据集,为正在做超分的同学提供数据集思路。

1 图像超分辨率数据集

1.1 DIV2K数据集

DIV2K数据集是用于图像超分辨率最流行的数据集之一,是为 NTIRE2017 和 NTIRE2018 超分辨率挑战赛收集的。该数据集由800张训练图像,100张验证集图像和100张测试集图像组成,每张图像都具有2K分辨率。

图片超分和视频超分数据集整理-StubbornHuang Blog

数据集官方下载地址:https://data.vision.ee.ethz.ch/cvl/DIV2K/

1.2 Flickr2K数据集

Flickr2K数据集同样是图像超分辨率最流行的数据集之一,包含2650张2k分辨率的图像,主要用于模型训练。

数据集官方下载地址:https://cv.snu.ac.kr/research/EDSR/Flickr2K.tar

1.3 DF2K数据集

DF2K数据集是DIV2K和Flickr2K两者的合并数据集,是《Enhanced Deep Residual Networks for Single Image Super-Resolution》 提出的。

可以分别下载DIV2K数据集和Flickr2K数据集,然后将两者进行合并。

1.4 DF2K+OST数据集

DF2K+OST是《Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Data》 提出的DF2K+OST的合并数据集。

OST是在《Recovering Realistic Texture in Image Super-resolution by Deep Spatial Feature Transform》 中提出的数据集,包含天空、山脉、植物、草地、水、动物和建筑物七个类别,图片分辨率在1k-2k之间,训练集包含10324张图片,测试集由300张图片组成。

图片超分和视频超分数据集整理-StubbornHuang Blog

OST数据集官方下载地址:https://mmlab.ie.cuhk.edu.hk/projects/SFTGAN/

1.5 LIU4k v2数据集

LIU4K-v2是一个4K超高清超分数据集,包含了1600张高分辨率训练图像和400张高分辨率验证图像,数据集中的图像分辨率至少为 3K,大部分属于4K-6K,包含非常多样化的背景和物体,并且低级信号分布也多样而复杂。

图片超分和视频超分数据集整理-StubbornHuang Blog

数据集官方下载地址:https://structpku.github.io/LIU4K_Dataset/LIU4K_v2.html

1.6 FFHQ数据集

Flickr-Faces-HQ是高质量的人脸图像数据集,由70000张分辨率为1024x1024的高质量人脸图像组成。

FFHQ数据集官方下载地址:https://github.com/NVlabs/ffhq-dataset

通用的测试基准数据集

图像超分的测试基准数据集通常包括:Set5、Set14、Urban100、BSDS100、Manga109、RealSR等数据集。

  • Set5数据集由5张图像组成(“baby”、“bird”、“butterfly”、“head”、“woman”)
  • Set14数据集由14张图像组成 (“baboon”, “barbara”, “bridge”, “coastguard”, “comic”, “face”, “flowers”, “foreman”, “lenna”, “man”, “monarch”, “pepper”, “ppt3”, “zebra”)
  • Urban100 数据集包含 100 个城市场景的建筑图像
  • BSDS100 数据集包含 100 个图像,范围从自然图像到植物、人物、食物等特定对象
  • Manga109 数据集包含 109 卷日本漫画
  • RealSR 数据集包含20张真实的从现有数据集选择或者从互联网下载的低分辨率图像

这些数据集可以从这里下载:https://drive.google.com/drive/folders/1B3DJGQKB6eNdwuQIhdskA64qUuVKLZ9u

2 视频超分辨率数据集

2.1 REDS数据集官方下载地址:

REDS数据集是在 NTIRE19 挑战赛中提出的,通常用于视频去模糊和视频超分辨率。该数据集由 240 个用于训练的视频序列、30 个用于验证的视频序列和 30 个用于测试的视频序列组成。每个视频序列都有 100 个连续帧,分辨率为 720x1,280。

REDS数据集官方下载地址:https://seungjunnah.github.io/Datasets/reds.html ,对于视频超分辨率任务,主要下载train_sharp、train_sharp_bicubic、val_sharp、val_sharp_bicubic等数据集。

2.2 Vimeo-90K数据集

Vimeo-90K是在Video Enhancement with Task-Oriented Flow 中提出的大规模高质量视频数据集。它专为以下四种视频处理任务而设计:时间帧插值、视频去噪、视频去块和视频超分辨率。该数据集由Triplet数据集(用于时间帧插值)和Septuplet数据集(用于视频去噪、去块和超分辨率)组成。 septuplet 数据集由 91,701个7帧序列组成,分辨率为 256x448。

Vimeo-90K数据集官方下载地址为:http://toflow.csail.mit.edu/ ,对于视频超分辨率任务,应该下载Septuplet数据集。

2.3 Vid4数据集

Vid4 数据集是用于视频超分辨率的最流行的测试数据集之一。该数据集由四个视频序列组成:“日历”(41 帧,分辨率为 576x720)、“城市”(34 帧,分辨率为 576x704)、“树叶”(49 帧,分辨率为 480x720)和“步行” '(47 帧,分辨率为 480x720)。

Vid4数据集下载地址:https://drive.google.com/file/d/1ZuvNNLgR85TV_whJoHM7uVb-XW1y70DW/view

2.4 SPMCS数据集

SPMCS数据集由 30 个视频序列组成。每个视频序列都有 31 个连续帧,分辨率为 540x960。每个视频序列包含 x2、x3 和 x4 比例因子的双三次下采样输入和高分辨率GT图像。

SPMCS数据集下载地址:https://github.com/jiangsutx/SPMC_VideoSR