[发明专利]一种基于darknet框架的物体检测系统

说明书

技术领域

本发明涉及一种检测系统，具体是一种基于darknet框架的物体检测系统。

背景技术

物体识别是计算机视觉中的一个研究方向，也是当前比较热门的研究领域。在人们的需求不断增长的今天，物体识别在安全、科技、经济方面正在起着举足轻重的作用，安防领域和交通监管部门也对物体识别提出了迫切的要求，所以研究物体识别对社会的未来有非常重要的意义。

传统目标检测系统采用deformable parts models(DPM)方法，通过滑动框方法提出目标区域，然后采用分类器来实现识别。近期的R-CNN类方法采用region proposal methods，首先生成潜在的边界盒，然后采用分类器识别边界盒所包含物体的类别。最后通过后处理方式来去除重复边界盒来进行优化。这类方法流程复杂，存在速度慢和训练困难的问题。

发明内容

本发明要解决的技术问题是提供一种基于darknet框架的物体检测系统，以解决背景技术中提到的问题。

为实现上述目的，本发明提供如下技术方案：

一种基于darknet框架的物体检测系统，包括

模型训练单元，用于设计darknet网络模型,darknet网络模型为卷积层和池化层的交替连接，对训练参数(包括学习率、)进行调整，利用数据集训练得到模型的权重参数；

预处理单元，用于收集合适的图片集，对图片集中的每一张图片进行物体标注，然后将标注信息转换成darknet训练所需的标注格式，并按照一定的比例生成相应的训练、验证、测试图片集；和定位识别单元所述；

定位识别单元；对预处理单元收集到的图片进行识别处理；模型训练单元分别连接预处理单元和定位识别单元。

作为本发明的优选方案：所述定位识别单元的识别处理步骤具体是：首先将图像分割成S*S的格子，每个格子预测B个边界盒和边界盒的置信值C，置信值代表box包含一个目标的置信度，如果没有目标，置信值为零。每一个边界盒包含5个值：x，y，w，h和confidence。(x，y)代表与格子相关的盒子的中心。(w，h)为与全图信息相关的盒子的宽和高。confidence代表预测盒子的IOU和ground truth，最终输出S*S*(B*5+C)维向量。

作为本发明的优选方案：所述向量包含对图片预测的边界盒坐标信息，置信值以及每个格子的类别概率，每个边界盒的置信分数通过对应格子的类别概率和盒子的置信度相乘得到，该分数衡量了该类别出现在盒子中的概率以及该盒子与目标的吻合程度。针对网络的输出框信息和类别概率，对其进行非极大抑制处理和阈值处理舍弃重复框，得到最终的检测结果。

与现有技术相比，本发明的有益效果是：本系统模型训练简单，整个系统是端到端(输入图片，输出检测框)的，并且可以实现实时检测。该系统采用全图信息来进行预测。与滑动窗口方法和region proposal-based方法不同，网络模型在训练和预测过程中可以利用全图信息，对目标框的定位更加准确。系统可以学习到目标的概括信息，并具有一定普适性。采用自然图片训练网络模型，然后采用艺术图像来预测，与比其它目标检测方法而言准确率高很多。该系统可以适应于不同的应用场景和检测目标，只需要用相应的数据集对系统进行训练即可。

附图说明

图1是一种基于darknet框架的物体检测系统和交互平台结构框图。

图2是卷积神经网络结构的示意图。

图3是一种基于darknet框架的物体检测系统和交互平台流程图。

具体实施方式

下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。

请参阅图1-3，一种基于darknet框架的物体检测系统，其包括预处理单元、模型训练单元和定位识别单元。

预处理单元：此步骤主要包括收集合适的图片集，对图片集中的每一张图片进行物体标注，然后将标注信息转换成darknet训练所需的标注格式，并按照一定的比例生成相应的训练、验证、测试图片集。

模型训练单元：设计darknet网络模型,darknet网络模型为卷积层和池化层的交替连接(具体可见2.2.2部分)，对训练参数(包括学习率、)进行调整，利用数据集训练得到模型的权重参数。