[发明专利]基于弱监督协同学习算法的安全帽识别方法及存储介质

摘要

本发明属于深度学习和目标识别技术领域，具体涉及一种基于弱监督协同学习算法的安全帽识别方法及存储介质。本发明的方法包括以下步骤：采用图像级别标签图像对弱监督协同学习算法网络进行训练；将待检测图像输入训练好的所述弱监督协同学习算法网络进行目标检测，得到概率向量；根据所述概率向量判定所述待检测图像中人物是否正确佩戴安全帽。本申请的技术方案采用一种新的协同学习构架，它在弱监督的学习过程中把弱监督检测子网络和有监督检测子网络连接成一个统一的整体，利用预测一致性损失加强了这两个检测子网络的实例预测一致性，这样让这个弱监督协同学习算法既具有弱监督的高效的训练网络的能力，也具有有监督的算法精准的检测精度。