发明公开
- 专利标题: 一种基于度量学习的安全服检测方法及系统
-
申请号: CN202211636645.5申请日: 2022-12-15
-
公开(公告)号: CN115775375A公开(公告)日: 2023-03-10
- 发明人: 于俊清 , 龚冬冬 , 王舒心 , 汤文炜
- 申请人: 国能长源武汉青山热电有限公司 , 华中科技大学
- 申请人地址: 湖北省武汉市青山区苏家湾;
- 专利权人: 国能长源武汉青山热电有限公司,华中科技大学
- 当前专利权人: 国能长源武汉青山热电有限公司,华中科技大学
- 当前专利权人地址: 湖北省武汉市青山区苏家湾;
- 代理机构: 华中科技大学专利中心
- 代理商 王颖翀
- 主分类号: G06V20/52
- IPC分类号: G06V20/52 ; G06V10/774 ; G06V10/82 ; G06N3/08 ; G06N3/0464
摘要:
本发明公开了一种基于度量学习的安全服检测方法及系统,该方法采用工作人员图像数据集训练TransReID模型;将待检测的工作人员图像帧进行目标检测后输入至训练好的TransReID模型,得到待分类特征图;从目标特征库中确定出与所述待分类特征图距离最近的前M个特征图,确定所述前M个特征图各自的安全服类型;将所述前M个特征图中频次最高的安全服类型作为所述待检测的工作人员图像帧的安全服类型检测结果。采用了度量学习中提取特征、计算距离得到目标分类的方式,使用TransReID神经网络作为特征提取网络提取工作人员衣物的特征,使用了自注意力机制,可以更好地提取特征。