[发明专利]特定着装人物的检测方法和装置

说明书

本发明提供一种特定着装人物的检测方法和装置。该方法包括：在训练特定着装人物分类的目标检测模型时，将该目标检测模型分成人物检测模型和第一着装分类模型，利用现有的人物检测模型对第一训练集合中的第一训练图片进行人物检测，并标注检测到的人物的边框，显示带有人物的边框的第一训练图片，用户对边框中的人物的第一着装类别进行标注，得到第二训练集合，第二训练集合中的第二训练图片包括人物的边框和着装类别，使用第二训练集合进行模型训练，得到第一着装分类模型，使用第一着装分类模型对第一待检测图片中的人物的着装进行检测。使得训练模型的过程节省了人力成本，同时，节约了训练的时间，提高了模型训练的效率。

技术领域

本发明涉及计算机视觉技术领域，尤其涉及一种特定着装人物的检测方法和装置。

背景技术

对于特定着装的人物的检测，可以简单快速地进行特定人物检索，人物追踪等；同样，工厂、电厂等场所需要入内人员穿着特定的着装，以保证入内人员以及工厂、电厂的安全。传统的对于特定着装人物的检测一般采用全人工查看的方法，这样需要耗费很多人力资源。

现有方案中，对特定着装人物进行检测，可以训练一个目标检测模型，使用该目标检测模型对图片中的特定着装人物进行检测。在训练目标检测模型时，需要在训练数据集上进行全人工标注，使用标注后的数据集训练目标检测模型，最后得到目标检测模型。

现有的训练目标检测模型的方法人工标注工作量很大，增加了人力成本。

发明内容

本发明提供一种特定着装人物的检测方法和装置，用以解决现有的训练目标检测模型的方法人工标注工作量很大，人力成本高的问题。

第一方面，本发明提供一种特定着装人物的检测方法，包括：

使用人物检测模型对第一训练集合中的第一训练图片进行人物检测，并标注检测到的人物的边框；

显示带有人物的边框的所述第一训练图片；

接收用户输入的所述边框中的人物的第一着装类别，得到第二训练集合，所述第二训练集合中的第二训练图片包括人物的边框和着装类别；

使用所述第二训练集合进行模型训练，得到第一着装分类模型；

使用所述第一着装分类模型对第一待检测图片中的人物的着装进行检测。

可选的，所述显示带有人物的边框的所述第一训练图片之后，还包括：

接收用户对所述第一训练图片中包括的人物的边框的调整操作；

根据所述调整操作调整所述第一训练图片中包括的人物的边框。

可选的，使用所述第一着装分类模型对待检测图片中的人物的着装进行检测之后，还包括：

显示分类检测结果，所述分类检测结果中包括所述待检测图片中的人物的着装类别；

接收用户对所述待检测图片中的人物的着装类别的反馈信息，所述反馈信息用于表示所述分类检测结果正确或者错误；

当分类检测结果错误的图片数量达到预设数量时，将所述分类检测结果错误的图片添加到所述第二训练集中；

使用更新后的所述第二训练集重新进行训练，得到更新后的所述第一着装分类模型。

可选的，所述第二训练集合中包括人物的着装类别为多种，所述第一着装分类模型为多分类着装分类模型，所述多分类着装分类模型用于检测多种着装类别。

可选的，所述方法还包括：

使用所述人物检测模型对第三训练图片进行人物检测，得到第三训练集合，所述第三训练集合中的第二人物检测结果图片包括人物的边框；

显示所述第二人物检测结果图片；