深度学习【56】物体检测:YOLOv3: An Incremental Improvement

YOLOv3在YOLOv2基础上进行了改进,主要体现在多尺度特征图预测和特征抽取网络。通过3个不同尺度的特征图预测提高小目标检测效果,借鉴FPN网络设计。特征抽取网络采用多尺度卷积核和跳跃连接。试验结果显示YOLOv3在速度和精度上表现良好,尤其是在一阶段检测框架中。
摘要由CSDN通过智能技术生成

YOLOv3的主要改进主要在两个方面,分别是分类网络的设计以及加入了多尺度特征图预测。当然还有其他的改变,比如进行多分类的时候不用softmax了而是使用independent logistic classifiers,然后损失函数是 binary cross-entropy loss。作者还是修复了yolov2中的数据读取bug,这提升了2%的mAP。

多尺度特征图预测

YOLOv3使用了3个不同尺度的特征图进行预测,来缓解YOLO对小目标检测的缺陷。借鉴于FPN网络的设计思想,YOLOv3在基础网络后面加入了一系列的上采样模块和卷积模块,来生成多尺度特征图用于预测。每个特征图只预测3个box,不像以前v2的5个box,这样总共预测9个box。这9个box的初始先验宽高也是用聚类聚出来的。

特征抽取网络

这里写图片描述

主要是使用了多尺度卷积核和跳跃连接。

具体的多尺度预测结构和特征抽取网络结构还是看作者提供的网络结构配置文件会清晰一点。
作者还尝试了一些失败的试验,比如加入focal loss,或者使用rcnn的样本选取策略。

试验结果

首先是分类网络在imagenet上的比较,速度和精度都不错:
这里写图片描述
与其他检测框架在精度上的比较,相比最好的二阶段框架还是差了些,但是对于其他的一阶段还是很好的:
这里写图片描述

速度的比较:
这里写图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值