该存储库包含开发的方法的源代码
此代码是从和修改的
网络结构
UWGAN用于生成逼真的水下图像
UWGAN将彩色图像及其深度图作为输入,然后通过生成对抗性训练学习参数,从而基于水下光学成像模型合成水下真实图像。 您可以在找到更多详细信息。
通过UWGAN合成的水下样式图像。 (a)是空中样本图像,(b)-(d)是不同水类型的合成水下样式样本图像。
基于UNet的水下图像恢复
拟议的用于水下图像恢复和增强的U-net体系结构。 比较了U-net中不同损失函数的影响,在比较的基础上提出了最适合水下图像恢复的损失函数,可以在找到有关损失函数的更多详细信息。
数据集
下载数据:
空中RGBD数据: ,
水下图片:[] [ ]
UIEB数据集进行验证:[ ]
我们用来训练UWGAN的NYU数据集:[] [ ]
从UWGAN生成的假水图像:[ ]
预训练模型:[ ]
UWGAN中的
2021-12-29 20:33:34
24.61MB
系统开源
1