[发明专利]一种基于多视角注意力机制的灰度图像自动上色方法

摘要

本发明涉及一种基于多视角注意力机制的灰度图像自动上色方法，属于灰度图像上色技术领域。通过步骤1，数据收集与处理，步骤2，模型构建，对特征图F进行全局平均池化，将特征图F进行通道池化，将输入特征图F不做任何变换成为Fout_3；有益效果在于，采用多视角注意力机制，多视角注意力模块能够让上色模型自适应地获取不同视角、不同尺度的注意力信息，更好地支持模型重建彩色图像。能够自适应地学习不同视角的特征信息，提高了编码‑解码结构的特征提取能力，强化了图像内语义信息和上下文信息的建模。通过将该模块嵌入编码‑解码结构，在保证上色结果图像质量的同时提高了颜色的饱和度和多样性。