一种基于度量学习的安全服检测方法及系统
摘要:
本发明公开了一种基于度量学习的安全服检测方法及系统,该方法采用工作人员图像数据集训练TransReID模型;将待检测的工作人员图像帧进行目标检测后输入至训练好的TransReID模型,得到待分类特征图;从目标特征库中确定出与所述待分类特征图距离最近的前M个特征图,确定所述前M个特征图各自的安全服类型;将所述前M个特征图中频次最高的安全服类型作为所述待检测的工作人员图像帧的安全服类型检测结果。采用了度量学习中提取特征、计算距离得到目标分类的方式,使用TransReID神经网络作为特征提取网络提取工作人员衣物的特征,使用了自注意力机制,可以更好地提取特征。
0/0