一种无监督的水下图像增强方法及相关设备

    公开(公告)号:CN115660980A

    公开(公告)日:2023-01-31

    申请号:CN202211296249.2

    申请日:2022-10-21

    摘要: 本发明公开了一种无监督的水下图像增强方法及相关设备,所述方法包括:在公开数据集中获取水下图像数据,所述水下图像数据包括训练数据集和测试数据集,将所述水下图像数据进行归一化处理和缩放处理;构建神经网络模型,使用处理后的所述训练数据集来训练所述神经网络模型,所述神经网络模型包括参数估计网络和图像增强网络;将处理后的所述测试数据集输入到已训练好的所述神经网络模型中,所述参数估计网络和所述图像增强网络根据处理后的所述测试数据集输出水下增强图像。本发明通过图像增强将失真的水下图像恢复为颜色均衡、细节丰富、对比度好的图像,实现了通过无监督的方法增强不同水体环境下的图像数据,泛用性更好。