[发明专利]一种高性能行人检索与重识别方法及装置

说明书

本发明公开了一种高性能行人检索与重识别方法及装置，该方法包括：分别获取单视角和多视角现实监控场景下的行人数据，对行人数据进行数据标注，其中单视角行人数据与COCO数据集中的行人部分，共同构建行人检测数据集，多视角行人数据构建行人重识别数据集；利用行人检测数据集基于Ghost轻量化模型改进的YOLOv5行人检测算法训练网络模型；利用行人重识别数据集训练得到行人重识别模型；搭建行人搜索系统。本发明通过深度模型压缩和算法算力的协同优化，从算法到硬件的自上而下的方法来优化深度学习的效率，实现低代价高性能的行人重识别系统。

技术领域

本发明属于行人检测和行人重识别技术领域，具体涉及一种高性能行人检索与重识别方法及装置。

背景技术

目标检测技术主要用于在给定图像中找到图像或视频中特定类别的对象，同时检测出目标的类标签及坐标。基于深度卷积神经网络算法成为目标检测领域的主流算法，根据不同的分类标准，目前可分为Two-stage算法和One-stage 算法。R-CNN系这类Two-stage算法，先生成候选框，然后对候选框分类并微调其位置。One-stage算法例如Yolo,SSD,则不需要预先生成候选框，直接对图像各个位置进行回归和分类。Two-stage算法精度较高，但速度较慢，算法改进伴随速度提升，One-stage算法速度快，算法改进伴随精度提升。

行人重识别技术是长时、跨域多目标跟踪的核心技术，主要目标是对同一个行人进行跨摄像头的再识别。现有的视频分析系统是多种任务算法的集合，对计算芯片的要求极高。行人搜索包含了查找(目标检测)和匹配(行人重识别)的过程。行人重识别模型，都是基于对监控视频进行预处理得到的行人图像，这需要大量的准备工作，针对现实场景，并且单独的行人重识别模型并不能满足应用要求。

不同于学术数据集单一场景，由于行人搜索算法将行人检测算法的结果图片输入到行人重识别模块，所以行人检测模型的检测效果，是识别正确行人的关键步骤。对于大型的视频监控系统，往往需要高性能、低功耗硬件的智能设备平台。行人重识别任务中，基于深度学习的方法是目前最好的算法。然而，复杂的深度学习模型通常具有较好的检测效果和重识别能力，难以在确保准确性和实时性的同时，部署在硬件资源有限且功率预算紧张的设备上。

发明内容

针对上述现有技术的不足，本发明提供一种高性能行人检索与重识别方法；模型在部署过程中主要面对模型大小，运行占用内存，计算效率等难点，因此本发明的核心就是对检测模型插入Ghost轻量化模块，并采取通道级稀疏化剪枝方法，从而使原有搜索网络模型大小缩小，准确度基准相当。并将训练后的模型于算丰SC5和计算卡上实施量化操作，轻量且快速地部署到硬件上，满足大型的视频监控系统需要高性能、低功耗硬件的智能设备平台的需求，确保准确性和实时性的同时，部署在硬件资源有限且功率预算紧张的设备上。

本发明的目的是通过以下技术方案实现的：

根据本说明书的第一方面，提供一种高性能行人检索与重识别方法，包括以下步骤：

S1，分别获取现实监控场景下的单视角行人数据和多视角行人数据，其中，采用单视角行人数据和COCO数据集中的行人部分数据共同构建行人检测数据集，多视角行人数据构建行人重识别数据集；

S2，利用S1中的行人检测数据集，基于Ghost轻量化模型改进的YOLOv5 行人检测算法训练行人检测模型；

S3，利用S1中的行人重识别数据集训练通道级稀疏化剪枝后的行人重识别网络，得到行人重识别模型；

S4，利用S2中训练好的行人检测模型和S3中的行人重识别模型基于算丰 SC5和云端AI计算加速卡进行量化部署，搭建行人搜索系统。

进一步地，步骤S2中的行人检测模型包括四种模块，分别为输入端模块、主干网络模块、颈部网络模块和输出端模块，输入为行人检测数据集中一张图片；