|
6 r) v4 v/ J* V1 f; [' o/ K6 m
有知友在我专栏文章下留言问数据集的事,那我就总结一下目前可以用于训练水下图像增强和测试水下图像增强深度学习模型的数据集吧。非成对的数据集比如U45,现在的使用价值有限了,本文重点关注成对数据集。 ' O9 g* k9 G1 e) N' D
ps:请大家自行下载,下载不了的话找代下服务吧,基本每天都有伸手党私聊我要数据集 我天天忙着攒pub,哪有精力一个一个发啊,链接都帮大家整理完了 大家挂个梯子自己下吧! ) M4 C. @+ M9 m. e4 R5 O+ @
UIEB: lichongyi大佬等开源的成对数据集,原图来自于水下真实图像,groundtruth来自于多种传统方法择优而取。890张大图和90张小图,小图测试起来非常方便,常用。
$ Z# O) B( g+ g [/ b2 y SUID:IEEE resource上开源。groundtruth是陆地上的真实图像,原图是用算法模拟出来的。
6 M: Y' Z; S+ p8 A( n: _ UFO120: 比较新的数据集,分为训练数据集和测试基准数据集,可以用于针对水下图像超分的模型训练任务。 + R: G+ o [( [5 e
EUVP:FUniE-GAN作者开源的数据集,很大,分成三个子集。采用七种不同的摄像设备采集水下图像,并从youtube视频上截取部分图像。Groundtruth是由训练好的CycleGAN生成。 9 s8 k1 ^: }5 ]( |% [. \
EUVP/UIEBD数据集: * Q+ r1 X5 y: e% J* e
7 G1 b, ] j% }3 X0 [
SUID数据集: 8 ^9 F2 ?5 m! s+ l4 F% c4 g+ R: ^
6 J- T2 O' L# D' V8 o0 f5 ?. o6 V
UFO120: - t6 D2 j, D' N5 Q* ~
J3 M' @2 t" a; m
$ @, t* S$ C9 L% `2 j4 z: T' q6 S" u' }$ w" ^
) Y- [6 }0 W) {; Q+ [! e* q2 y3 d0 M/ @& d7 E. S1 {0 O' y) p
|