[发明专利]车载图像目标检测方法、系统及计算机可读存储介质

说明书

本发明公开了一种基于车载AI芯片的城市目标检测的深度学习方法、系统及计算机可读存储介质，涉及计算机视觉中的目标检测领域。方法包括以下步骤：对训练后的所述深度卷积神经网络进行模型转换与量化处理，生成车载端芯片支持的WK格式文件；车载端芯片读取、加载所述WK格式文件，并初始化NNIE对象；读取待检测的目标图像，使用所述NNIE对象对所述目标图像进行转化并将数据转移到所述车载端芯片中进行推理，获得推理的多维向量数组；对所述多维向量数组进行NMS处理，去除冗余框，得到最终结果。可以解决现有的车载目标检测方法对于复杂场景，难以满足实时使用的问题。

技术领域

本发明涉及计算机视觉中的目标检测领域，具体是涉及一种车载图像目标检测方法、系统及计算机可读存储介质。

背景技术

随着社会经济的发展及城市化进程的加快，机动车保有量逐年激增，行人运动轨迹复杂，导致城市道路承载能力逐渐无法满足机动车无限增长的需求，交通拥堵、交通违章以及交通事故等问题亦随之不断衍生。从交通管理部门的人力资源配置与行车安全角度出发，一方面，车辆与行人的智能检测与感知以及行为分析在保证检测准确率与实时性的同时，可以极大程度地减少交通管理部门人力资源的浪费以及工作人员的压力；另一方面，智能化检测方法的不断提高，可以有效地阻止车祸发生率。从规范化管理与合理调度的角度出发，车辆与行人的智能检测可以获得车辆的类别、位置信息，并依据该信息进一步计算交通流量，分析当前路段的行车状况以及估计车辆行为以方便管理部门从全局的视角统一管理、合理调度，从而保证道路的通行能力。长久以来，车辆的检测与分析之所以成为智能交通领域的研究热点与难点，正是由场景的复杂性以及车辆行为的随机性所导致。雨、雾、拥堵、光照等都会造成车辆的漏检率高且鲁棒性差，为了提高复杂场景下车载车辆检测的准确性、实时性与鲁棒性，对交通管制、行为分析以及智能城市管理具有重要的研究价值，实现通用可移植的车载端边缘计算的智能车辆以及行人的目标检测是一个十分迫切的需求。

发明内容

本发明的目的是为了克服上述背景技术的不足，提供一种车载图像目标检测方法、系统及计算机可读存储介质，用于解决现有的车载目标检测方法对于复杂场景检测过程中出现的目标重度遮挡、快速移动、大雨大雾恶劣天气等情况下，容易导致检测准确率低、虚警率高、检测时间长、难以满足实时使用的问题。

第一方面，提供一种车载图像目标检测方法，包括以下步骤：

使用测试场景相同的数据作为训练集，设定检测的目标类别数，对预设的深度卷积神经网络进行训练；

对训练后的所述深度卷积神经网络进行模型转换与量化处理，生成车载端芯片支持的WK格式文件；

车载端芯片读取、加载所述WK格式文件，并初始化NNIE对象；

读取待检测的目标图像，使用所述NNIE对象对所述目标图像进行转化并将数据转移到所述车载端芯片中进行推理，获得推理的多维向量数组；

对所述多维向量数组进行NMS处理，去除冗余框，得到最终结果。

在第一方面的一种可能实施例中，所述深度卷积神经网络包括基于darknet53的YOLO V3神经网络模型。

在第一方面的一种可能实施例中，所述对训练后的所述深度卷积神经网络进行模型转换与量化处理的步骤包括：

将训练好的YOLO V3神经网络模型利用darknet2caffe工具，转换为caffemodel；

利用RuyiStudio工具，对caffemodel进行量化操作，将所述YOLO V3神经网络模型量化为所述车载端芯片支持的8bit的WK格式文件。

在第一方面的一种可能实施例中，所述对所述多维向量数组进行NMS处理，去除冗余框，得到最终结果的步骤包括：