一种基于多通道介质传输先验的水下图像增强方法及设备
申请号:CN202410793079
申请日期:2024-06-19
公开号:CN118823560A
公开日期:2024-10-22
类型:发明专利
摘要
本发明涉及一种基于多通道介质传输先验的水下图像增强方法及设备,该方法包括:获取水下图像数据集,获取水下图像的深度图,将水下图像的RGB三通道进行分离,结合深度图估计每个通道的介质传输图,根据介质传输图构建基于多通道介质传输先验的生成式对抗网络,将获取的水下图像数据集输入训练好的生成式对抗网络,实现水下图像自适应增强。与现有技术相比,本发明具有提高水下图像的增强性能和鲁棒性等优点。
技术关键词
水下图像增强方法
生成式对抗网络
水下图像数据
多通道
深度图
介质
注意力先验
深度估计方法
编码器
输出模块
预训练模型
空洞
上采样
存储器
鲁棒性
解码器