买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：普天信息技术有限公司

摘要：本发明实施例提供一种电力作业人员安全着装识别方法及装置，所述方法包括：采集电力作业人员着装图像；将所述电力作业人员着装图像输入至预置的着装识别模型中进行人员头部和身体的检测以及安全帽分类识别和工装分类识别，获得安全帽识别结果和工装识别结果。本发明实施例提供的电力作业人员安全着装识别方法及装置，采用轻量级网络MobileNet作为基础特征提取网络，结合优化的SSD目标检测算法，对人员头部与身体部位初步判定，然后将网络不同层次特征、不同尺度拼接特征应用于不同分类识别任务，降低计算量同时，提升对安全着装分类识别性能。

主权项：1.一种电力作业人员安全着装识别方法，其特征在于，包括：采集电力作业人员着装图像；将所述电力作业人员着装图像输入至预置的着装识别模型中进行人员头部和身体的检测以及安全帽分类识别和工装分类识别，获得安全帽识别结果和工装识别结果；其中，所述着装识别模型是预先以电力作业人员着装图像为样本，以所述电力作业人员着装图像中人员戴安全帽和穿工装情况为样本标签进行训练获得，所述着装识别模型包括MobileNet网络、SSD目标检测模块、安全帽分类识别模块和工装分类识别模块；所述将所述电力作业人员着装图像输入至预置的着装识别模型中进行人员头部和身体的检测以及安全帽分类识别和工装分类识别，获得安全帽识别结果和工装识别结果的步骤，具体包括：将所述电力作业人员着装图像输入至所述MobileNet网络中，获得所述电力作业人员着装图像对应的基础特征；将所述电力作业人员着装图像对应的基础特征输入至所述SSD目标检测模块，获得预测目标框位置与大小，并基于所述预测目标框位置与大小，确定所述预测目标框内部为人员头部或人员身体；将内部为人员头部的预测目标框对应的基础特征输入至所述安全帽分类识别模块，获得安全帽识别结果，将内部为人员身体的预测目标框对应的基础特征作为浅层特征，将所述浅层特征输入至多个卷积层，获得深层特征，将所述浅层特征与深层特征拼接后输入至所述工装分类识别模块，获得工装识别结果；训练获得所述着装识别模型的步骤，具体包括：采集不同人员的不同着装图像，将每张着装图像中的人员戴安全帽情况和穿工装情况打上相应的标签，构建样本集，将所述样本集划分为训练样本集和测试样本集；利用所述训练样本集对所述着装识别模型进行训练，训练结束后，获得初步训练完成的着装识别模型；利用所述测试样本集对所述初步训练完成的着装识别模型进行测试，若测试精度满足预设要求，则获得最终训练完成的着装识别模型；或者，若测试精度没有达到预设要求，则针对未标注的着装图像，通过利用所述初步训练完成的着装识别模型的预测结果、结合阈值比较和非极大值抑制计算结果进行自动标注，并通过自动标注与人工微调相结合的方式，优化所述样本集；利用优化后的样本集对所述初步训练完成的着装识别模型进行迭代优化训练直至所述着装识别模型的测试精度满足所述预设要求；针对未标注的着装图像，通过利用所述初步训练完成的着装识别模型的预测结果、结合阈值比较和非极大值抑制计算结果进行自动标注，具体包括：将所述未标注的着装图像输入至所述初步训练完成的着装识别模型，获得若干个预测结果，每个预测结果包括预测目标框的位置、预测目标框所属类别与预测目标框的分数；将所有预测目标框的分数与所设定的分数阈值进行比较，若某一预测目标框的分数高于所述分数阈值，将所述预测目标框所对应的位置、大小与类别信息作为标注信息；利用预测目标框的分数低于所述分数阈值的所有预测结果构造一个集合，若所述集合中存在N个预测目标框属于同一区域，且N小于所设定的目标框数目阈值，则丢弃所述预测结果，若N大于所设定的目标框数目阈值，则取N个预测目标框中大多数框所属的类别，记为M个框所属的类别，作为所述未标注的着装图像的类别属性，并利用所述M个框的位置与大小信息、采用非极大值抑制NMS计算的方式找到最佳的预测目标框位置与大小作为标住信息。

全文数据：

权利要求：

百度查询： 普天信息技术有限公司 电力作业人员安全着装识别方法及装置