向AI转型的程序员都关注了这个号????????????
大数据挖掘DT机器学习 公众号： datayx
全部工程已传到github
代码获取：
关注微信公众号 datayx 然后回复 目标检测 即可获取。
训练好的模型请在yolo2检测模型下载
https://pan.baidu.com/s/1ZeT5HerjQxyUZ_L9d3X52w
一、全部代码解读如下：
1、model_darknet19.py：yolo2网络模型——darknet19。
YOLOv2采用了一个新的基础模型（特征提取器），称为Darknet-19，包括19个卷积层和5个maxpooling层，如下图。Darknet-19与VGG16模型设计原则是一致的，主要采用3*3卷积，采用2*2的maxpooling层之后，特征图维度降低2倍，而同时将特征图的channles增加两倍。
主要特点有：
（1）去掉了全连接层fc
·这样大大减少了网络的参数，个人理解这是yolo2可以增加每个cell产生边界框以及每个边界框能够单独的对应一组类别概率的原因。
·并且，网络下采样是32倍，这样也使得网络可以接收任意尺寸的图片，所以yolo2有了Multi-Scale Training多尺度训练的改进：输入图片resize到不同的尺寸（论文中选用320,352...，608十个尺寸，下采样32倍对应10*10~19*19的特征图）。每训练10个epoch，将图片resize到另一个不同的尺寸再训练。这样一个模型可以适应不同的输入图片尺寸，输入图像大(608*608)精度高速度稍慢、输入图片小(320*320)精度稍低速度快，增加了模型对不同尺寸图片输入的鲁棒性。
（2）在每个卷积层后面都加入一个BN层并不再使用droput
·这样提升模型收敛速度，而且可以起到一定正则化效果，降低模型的过拟合。
（3）采用跨层连接Fine-Grained Features
·YOLOv2的输入图片大小为416*416，经过5次maxpooling（下采样32倍）之后得到13*13大小的特征图，并以此特征图采用卷积做预测。这样会导致小的目标物体经过5层maxpooling之后特征基本没有了。所以yolo2引入passthrough层：前面的特征图维度是后面的特征图的2倍，passthrough层抽取前面层的每个2*2的局部区域，然后将其转化为channel维度，对于26*26*512的特征图，经passthrough层处理之后就变成了13*13*2048的新特征图，这样就可以与后面的13*13*1024特征图连接在一起形成13*13*3072大小的特征图，然后在此特征图基础上卷积做预测。作者在后期的实现中借鉴了ResNet网络，不是直接对高分辨特征图处理，而是增加了一个中间卷积层，先采用64个1*1卷积核进行卷积，然后再进行passthrough处理，这样26*26*512的特征图得到13*13*256的特征图。这算是实现上的一个小细节。
2、decode.py：解码darknet19网络得到的参数.
3、utils.py：功能函数，包含：预处理输入图片、筛选边界框NMS、绘制筛选后的边界框。
这里着重介绍NMS中IOU计算方式：yolo2中计算IOU只考虑形状，先将anchor与ground truth的中心点都偏移到同一位置（cell左上角），然后计算出对应的IOU值。
IOU计算难点在于计算交集大小：首先要判断是否有交集，然后再计算IOU。计算时候有一个trick，只计算交集部分的左上角和右下角坐标即可，通过取max和min计算：
4、Main.py：YOLO_v2主函数
对应程序有三个步骤：
（1）输入图片进入darknet19网络得到特征图，并进行解码得到：xmin xmax表示的边界框、置信度、类别概率
（2）筛选解码后的回归边界框——NMS
（3）绘制筛选后的边界框
运行环境：
Python3 + Tensorflow1.5 + OpenCV-python3.3.1 + Numpy1.13
准备工作：
请在yolo2检测模型下载模型，并放到yolo2_model文件夹下
https://pan.baidu.com/s/1ZeT5HerjQxyUZ_L9d3X52w
文件说明：
1、model_darknet19.py：yolo2网络模型——darknet19
运行Main.py即可得到效果图：
1、car.jpg：输入的待检测图片
可以看到，跟yolo1对比，yolo2引入anchor后检测精度有了提升（car和person的类别置信度高了许多），并且每个边界框对应一组类别概率解决了yolo1中多个目标中心点落在同一个cell只能检测一个物体的问题（左侧两个person都检测出来了）。相比yolo1还是有一定提升的。
阅读过本文的人还看了以下：
不断更新资源
深度学习、机器学习、数据分析、python
搜索公众号添加： datayx
长按图片，识别二维码，点关注
