有效

一种水下图像增强模型训练方法及增强方法、设备及介质

沈礼权、余檬
上海大学

摘要

本发明涉及一种水下图像增强模型训练方法及增强方法、设备及介质,其中所述的模型包括水下内容编码器、水下失真编码器、空中内容编码器、合成图像生成器、增强图像生成器、分割编码解码网络、分割解码器和语义感知对比模块;该模型训练方法通过训练图像预处理、前向交叉转换、后向交叉转换、语义感知和双任务引导步骤获取用于计算模型损失的相关数据,利用模型损失对所述的水下图像增强模型进行训练;所述的增强方法利用上述训练方法训练后的模型实现对待增强水下图像进行增强处理。与现有技术相比,本发明提供一种基于解纠缠表示的模型训练方法,使得训练后的模型能专注于增强对机器视觉任务有益的特征从而达到提高水下机器视觉任务性能的目的。