[发明专利]一种改进U2

摘要

本发明公开了一种改进U2‑Net的显著性蒙古族图案检测算法，包括：获取待检测图像；将待检测图像输入显著性蒙古族图案检测模型，输出得到显著性蒙古族图案检测结果；其中显著性蒙古族图案检测模型为基于改进的U2‑Net网络的深度学习模型，包括编码器、解码器、特征补充提取模块、轻量级组合注意力模块和特征图融合模块；其中特征补充提取模块采用小尺寸卷积核串联的方式，能够增大感受野，减少计算成本，预测得到的显著图更加完整；轻量级组合注意力模块在不显著增加计算成本的前提下，使显著图的边界定位更加精准，同时抑制冗余的背景噪声，解决了显著性蒙古族图案的预测存在不完整、边缘不够精确、检测精确度不高的技术问题。