物体检测Yolov3_训练与Ascend310推理

【摘要】 一、概念介绍    Yolo是一种目标检测算法,将目标区域预测和目标类别预测整合于单个神经网络模型中,实现在准确率较高的情况下快速目标检测与识别。Yolov3是在v1和v2的基础上融合了一些好的方案,在保持速度优势的同时,提升了预测精度,尤其加强了对小物体的识别能力,更加适合现场应用环境。二、网络结构    Yolov3相对于v1和v2的主要改进是使用了新的网络结构Darknet-53,改进…

一、概念介绍

Yolo是一种目标检测算法,将目标区域预测和目标类别预测整合于单个神经网络模型中,实现在准确率较高的情况下快速目标检测与识别。Yolov3是在v1和v2的基础上融合了一些好的方案,在保持速度优势的同时,提升了预测精度,尤其加强了对小物体的识别能力,更加适合现场应用环境。

二、网络结构

Yolov3相对于v1和v2的主要改进是使用了新的网络结构Darknet-53,改进了模板框的预测。

上图红框标出了Darknet-53(含有53个卷积层),借鉴了残差网络做法,在一些层之间使用了shourtcut,如下图就是一个残差组件。

随着输出的特征图的数量和尺度的变化,先验框的尺寸也需要相应的调整。Yolov2已经开始采用K-means聚类得到先验框的尺寸,Yolov3延续了这种方法,为每种下采样尺度设定3种先验框,总共聚类出9种尺寸的先验框。先验框针对大、中和小物体有不同尺寸,适合检验不同大小的物体。

ModelArts的Yolov3_ResNet18没有对原版的Yolov3没有做精度和速度的优化,使用的主干网络是resnet18,比原版的darknet53提起特征的能力更弱点,只是适配了Ascend310的推理。

© 版权声明
THE END
喜欢就支持一下吧
点赞0 分享