基于注意力机制的轻量级水下图像风格迁移方法

王希雅, 王奔

杭州师范大学学报(自然科学版) ›› 2023, Vol. 22 ›› Issue (4) : 381 -388.

PDF
杭州师范大学学报(自然科学版) ›› 2023, Vol. 22 ›› Issue (4) : 381 -388. DOI: 10.19926/j.cnki.issn.1674-232X.2023.04.007

基于注意力机制的轻量级水下图像风格迁移方法

    王希雅, 王奔
作者信息 +

Author information +
文章历史 +
PDF

摘要

目前,获取海洋素材的途径稀少,数据集缺失,质量相对较低,是海洋生态环境保护宣传、水下图像应用设计的重大难题之一.本文将基于深度学习的图像风格迁移技术与白化现象严重的水下图像进行结合,引入SqueezeNext轻量级编码器网络,通过低秩卷积、压缩模块和全连接操作,在不影响模型准确率的条件下减少模型参数冗余,减少模型计算量.同时,本文还进一步引入了关系感知全局注意力转换网络,结合特征和特征位置的相关性更有效地平衡全局和局部结构信息.此外,在损失函数中加入HSV色相损失函数,通过色相值的计算来衡量颜色的距离,由此更好地表示颜色的相近关系.通过学习图像的内容、风格等特征进行照片风格迁移,解决海洋图像设计素材短缺、图像质量低等问题,从而增加水下图像样式的多元性、素材的丰富性以及资源的拓展性.

关键词

风格迁移 / 编码器-解码器 / 注意力机制 / 损失函数

Key words

引用本文

引用格式 ▾
基于注意力机制的轻量级水下图像风格迁移方法[J]. 杭州师范大学学报(自然科学版), 2023, 22(4): 381-388 DOI:10.19926/j.cnki.issn.1674-232X.2023.04.007

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

AI Summary AI Mindmap
PDF

0

访问

0

被引

详细

导航
相关文章

AI思维导图

/