一种基于改进U-Net网络架构的水下图像增强模型的构建方法
摘要:
本发明公开了一种基于改进U‑Net网络架构的水下图像增强模型的构建方法,该方法首先通过点卷积和深度卷积替换U‑Net网络的标准卷积,从而构建基于U‑Net网络的图像增强模型网络架构,该模型架构包括多个基本块和SK融合模块;然后构建基本块的结构;随后在构建的U‑Net网络基本块内部引入简化的通道注意力机制;最后构建SK融合模块的结构,完成模型网络架构的构建;解决了现有技术的模型对于复杂动态的水下环境假设不合理,导致图像过渡增强、过饱和度,从而影响图像增强质量的问题,具有在保证网络结构轻量化的前提下,设计的模型效果仍能超越现有技术的特点。
0/0