安全帽的检测方法、装置、计算机设备及存储介质

基本信息

申请号 CN202110811733.3 申请日 -
公开(公告)号 CN113516082A 公开(公告)日 2021-10-19
申请公布号 CN113516082A 申请公布日 2021-10-19
分类号 G06K9/00(2006.01)I;G06K9/32(2006.01)I;G06K9/62(2006.01)I;G06N20/00(2019.01)I 分类 计算;推算;计数;
发明人 钱青;胡辰;卜景德 申请(专利权)人 曙光信息产业(北京)有限公司
代理机构 北京品源专利代理有限公司 代理人 孔凡红
地址 100193北京市海淀区东北旺西路8号院36号楼
法律状态 -

摘要

摘要 本发明实施例公开了一种安全帽的检测方法、装置、计算机设备及存储介质,该方法包括:获取包括至少一个检测对象的目标检测图像;将目标检测图像输入至预先训练的检测模型中,并获取检测模型输出的,在目标检测图像中检测得到的至少一个检测对象的头部所在区域;根据检测模型输出的检测结果,在目标检测图像中分别截取各检测对象的头部区域图像;将各检测对象的头部区域图像分别输入至分类模型中,并获取分类模型输出的各检测对象是否佩戴安全帽,或者是否佩戴设定颜色的安全帽的分类结果。本发明实施例的技术方案可以有效提高安全帽的检测精度,在保证安全帽检测精度的基础上,同时可以提高对每个独立模型的训练和优化速度。