一种基于注意力机制的弱监督目标检测方法及系统
摘要:
本发明提供了一种基于注意力机制的弱监督目标检测方法及系统,包括:将获取的待检测图像通过卷积层进行特征提取,得到特征图;将所述特征图输入到注意力机制模块,得到空间增强后的特征图;基于所述空间增强后的特征图确定所述待检测图像是否为目标图像;其中,所述注意力机制模块,基于卷积层对特征图进行空间维度归一化得到注意力机制权重图,并结合所述特征图进行全局平均化操作得到空间增强后的特征图。本发明采用卷积神经网络结合注意力机制目标检测的方法,解决了高精度标注的数据集费时费力,提高了准确率,实现了通过低质量的数据集来获得高质量的目标检测模型。
0/0