参考图像指导下融合自注意力机制的废钢图像去尘方法及系统

摘要

本发明提出一种参考图像指导下融合自注意力机制的废钢图像去尘方法及系统，首先使用马氏距离和欧式距离在无尘参考图像及带尘图像上构建相似块组标签；随后定义距离度量和对比损失函数构建全卷积相似块组搜索网络。利用带标签的相似块组训练神经网络，获得与块匹配相关的特征；对带尘图像进行分块，针对每个带尘图像块，在学习到的特征指导下寻找相似块。带尘的相似块构成块组，融合自注意力机制，构建块组协同去尘网络。经过块组协同去尘得到干净块组，通过聚合图像块得到复原后清晰的无尘图像。