[发明专利]基于注意力机制的图像去马赛克方法

摘要

本发明公开了一种基于注意力机制的图像去马赛克方法，包括以下步骤：步骤构建数据集，包括训练集、验证集和测试集，并对数据集进行预处理，仿真得到Bayer图像；采集得到的数据分别进行卷积处理，然后提取特征；采用基于注意力机制的端到端神经网络模型，得到去马赛克模型；处理测试集，测试得到的模型性能；进行定量评价，评价指标采用峰值信噪比PSNR。本发明的优点是：本发明提供了一种基于注意力机制的图像去马赛克方法，该方法将注意力机制引入到图像去马赛克任务中，从众多信息中选择出对去马赛克任务目标更关键的信息，提高深度学习网络在去马赛克任务中的表现，使得生成图像更加接近原始真实图像。