前段时间看了YOLO的论文，打算用YOLO模型做一个迁移学习，看看能不能用于项目中去。但在实践过程中感觉到对于YOLO的一些细节和技巧还是没有很好的理解，现学习其他人的博客总结（所有参考连接都附于最后一部分“参考资料”），加入自己的理解，整理此学习笔记。
概念补充：mAP：mAP是目标检测算法中衡量算法精确度的一个指标，其涉及到查准率（Precision）和查全率（Recall）。对于目标检测任务，对于每一个目标可以计算出其查准率和查全率，多次实验进行统计，可以得到每个类有一条P-R曲线，曲线下面的面积即是AP的值，m的意思是对每一个类的AP值求平均，即mAP的定义，mAP的大小在[0, 1]区间内。
0 - 背景
过去的目标检测算法很多的思路是通过算法（经典的如region proposal）产生大量的可能包含目标物体的边界框（potential bounding box），然后用分类器去判断每一个边界框里面是否包含物体，以及包含物体的类别以及置信度（confidence），经典的模型由R-CNN以及其多个变种。但正如YOLO论文中反复提及的，这种做法一方面会使得训练变得麻烦（因为整个模型由多个部分组成，而每一部分通常都需要独立训练），不仅费时，也很难综合几部分结果求得最好的整体结果，另一方面，多个部分的协同使得整个模型的效率较为低下。
一直以来，计算机视觉对于检测两个距离很近的同类目标或者不同目标都具有很大的挑战。很多算法会对输入的图像进行尺寸的放缩，而如果当图像中由两个目标相近的时候，加上缩小尺寸，这使得对于相近目标的检测变得很困难。对于这种情况（特别是小目标检测，如鸟群），YOLO v3做得很好。
1 - 比较
YOLO不同于上述的目标检测算法，它将目标检测抽象为一个回归问题（regression problem），直接使用一个神经网络，输入为原始图片，输出为边界框坐标、目标类别、目标置信度等，YOLO将其他模型分开的部分都综合到一个神经网络中，使得训练能够采用端到端（end-to-end）来优化。同时，这种改进不仅使得训练变得容易，也使得检测速度很快。并且，论文中提到，YOLO迁移到其他数据集（如艺术图像）上的表现远好于其他目标检测算法，这表明了YOLO较其他模型学习了较好的抽象特征，泛化能力更强。
但YOLO也有缺点，其在物体定位时容易出错，不过在误将在背景检测出本不存在的物品的情况相对少一些。
不同算法的检测流程对比如下（图来自于博客：https://blog.csdn.net/hrsstudy）：
综上，将YOLO的优点总结如下：
- 检测快速
- 不同于其他检测算法，YOLO的模型只由一个神经网路构成，原始图像和最终结果是模型的输入和输出。论文中提到，YOLO在Titan X的GPU上能达到45 FPS，其快速版本Fast YOLO能达到155 FPS
- 减少背景错误（将背景识别成本不存在的物体）
- 其他检测算法，采用的诸如滑动窗口或者region proposal产生预选边界框，一定程度忽略了全局信息（上下文信息），而过多关注了局部信息。而YOLO采用整张图片进行输入，很好的利用了检测目标的上下文信息（环境信息），从而有效避免了背景错误（值得一提的是，YOLO的背景错误不到Fast R-CNN的一半）
- 较好的学习了目标的泛化特征
- YOLO对于其他数据集的迁移学习能力较强，这说明其学到了目标的泛化特征，提高了泛化能力。我的理解是，相比其他算法，YOLO用一个神经网络代替了复杂的多部分综合，从而减少了人工干预，使得模型能学习到更加好的参数，从而提高了效果
以上总结了YOLO的优点，下面总结其缺点：
- YOLO的目标检测精度低于R-CNN为代表的其他state-of-the-art的物体检测算法
- YOLO的定位错误相比其他算法较为严重
- YOLO对小物体的检测效果也不佳，这是因为划定网络单元的时候限制了其每一个单元最多打的边界框数量，因此网格单元的数量以及每个单元的边界框数量都成了额外影响结果的参数
下面是各个目标检测算法的性能对比（图来自于博客：https://blog.csdn.net/hrsstudy）：
2 - 相关计算
YOLO将图像分成$S \times S$个网格单元，每一个网格单元最后输出$B$个边界框（每个边界框包括$x,y,w,h,confidence$），以及$C$个类别的概率。因此，最后输出的张量维度为$S \times S \times (B \times 5 \add C)$。
