【综述】Pytorch YOLO项目推荐建议收藏学习

最近一些群友有询问我有没有哪些YOLO的算法推荐，考虑到现在Pytorch是做实验发论文最流行的深度学习框架，所以我就针对Pytorch实现的YOLO项目做了一个盘点和汇总，真心希望可以帮助到入门目标检测的同学。写作不易，要是觉得这篇文章有用就点个关注吧QAQ。

前言

这篇推文的目的是为了给目标检测领域的刚入门或者希望进一步提升自己的能力的同学做的一个YOLO项目推荐。推荐的项目都是当前流行，Star较高，用Pytorch框架实现的基于YOLO的检测项目，建议收藏和学习。

YOLOv3-model-pruning

项目地址：https://github.com/Lam1360/YOLOv3-model-pruning
项目特点：用 YOLOv3 模型在一个开源的人手检测数据集 oxford hand 上做人手检测，并在此基础上做模型剪枝。对于该数据集，对 YOLOv3 进行 channel pruning 之后，模型的参数量、模型大小减少 80% ，FLOPs 降低 70%，前向推断的速度可以达到原来的 200%，同时可以保持 mAP 基本不变。
原理简介：这个代码基于论文 ICLR 2017《Pruning Filters for Efficient ConvNets》进行改进实现的 channel pruning算法，类似的代码实现还有这个 yolov3-network-slimming（地址：https://github.com/talebolano...）。原始论文中的算法是针对分类模型的，基于 BN 层的 gamma 系数进行剪枝的。剪枝步骤就是稀疏训练->剪枝->微调。
剪枝结果：

下面是对部分卷积层进行剪枝前后通道数的变化：

部分卷积层的通道数大幅度减少

剪枝前后指标对比：

加入稀疏正则项之后，mAP 反而更高了（在实验过程中发现，其实 mAP上下波动 0.02 是正常现象），因此可以认为稀疏训练得到的 mAP 与正常训练几乎一致。将 prune 后得到的模型进行 finetune 并没有明显的提升，因此剪枝三步可以直接简化成两步。剪枝前后模型的参数量、模型大小降为原来的 1/6 ，FLOPs 降为原来的 1/3，前向推断的速度可以达到原来的 2 倍，同时可以保持 mAP 基本不变。需要明确的是，上面表格中剪枝的效果是只是针对该数据集的，不一定能保证在其他数据集上也有同样的效果。

总结：这个剪枝项目是可以在工程中进行应用的，在这之前如果你想学习这个算法的原理，那么推荐看我写的这篇：深度学习算法优化系列八 | VGG，ResNet，DenseNe模型剪枝代码实战

Slim YOLOV3

项目地址：https://github.com/PengyiZhang/SlimYOLOv3
项目特点：《SlimYOLOv3: Narrower, Faster and Better for Real-Time UAV Applications》是arXiv， 25 Jul 2019的论文，作者全部来自北理，论文链接：arxiv.org/abs/1907.1109。作者对YOLOv3的卷积层通道剪枝(以通道级稀疏化)，大幅削减了模型的计算量(~90.8% decrease of FLOPs)和参数量( ~92.0% decline of parameter size)，剪枝后的模型运行速度约为原来的两倍，并基本保持了原模型的检测精度。
原理介绍：这篇论文基本上就是ICLR 2017《Pruning Filters for Efficient ConvNets》这篇论文在YOLOV3-SPP上的一个应用。原理可以用下图解释：

在这里插入图片描述

算法结果：

结果1

结果2

论文展示效果：

可视化效果图

总结：说白了这篇论文就是剪枝算法在YOLOV3-SPP上的应用，技术含量其实也就那样。。。但是新手的话还是推荐大家来学习一下。

Anchor聚类 kmeans-anchor-boxes

项目地址：https://github.com/lars76/kmeans-anchor-boxes
项目特点：针对自己的数据集聚类出最适配的Anchor，可以加快收敛速度并有可能提升模型效果。
原理介绍：我之前写了一篇推文仔细的讲解这个，地址如下：目标检测算法之YOLO系列算法的Anchor聚类代码实战
在VOC 2007数据集上获得的算法结果：

论文中的不同聚类算法获得的平均IOU对比

在结果测试时，YOLOv2采用的5种Anchor可以达到的Avg IOU是61，而Faster-RCNN采用9种Anchor达到的平均IOU是60.9，也即是说本文仅仅选取5种Anchor就可以达到Faster-RCNN中9种Anchor的效果。

ASFF

论文地址：https://arxiv.org/abs/1911.09516
项目地址：https://github.com/ruinmessi/ASFF
项目特点：超强Tricks！
贡献：这篇论文最大的创新点是提出了一种新的特征融合的方式ASFF，通过学习权重参数的方式将不同层的特征融合到一起，作者证明了这样的方法要优于通过concatenation或者element-wise的方式。不仅如此，作者还在YOLOv3的基础上吸取了很多优秀的经验，构建了一个非常强劲的目标检测baseline，这个baseline的mAP就达到了38（之前的YOLOV3mAP值是33%左右，ORZ），其中包括：
Guided Anchoring
Bags of Tricks
Additional IoU loss
算法原理：这里不多介绍，之后会专门写一篇文章介绍一下，这里看一下论文的核心原理图：

ASPP

算法结果：

精度和速度对比

在COCO数据集的精度测试。

总结：ASPP这篇论文可以算是集百家之长，并且ASPP创新点也是拉满，让我第一次知道空间信息的叠加也是可学习的，结果也非常惊艳，推荐大家学习。

yolov3-channel-and-layer-pruning

项目地址：https://github.com/tanluren/yolov3-channel-and-layer-pruning
项目特点：本项目以ultralytics/yolov3为基础实现，根据论文Learning Efficient Convolutional Networks Through Network Slimming (ICCV 2017)原理基于bn层Gmma系数进行通道剪枝，下面引用了几种不同的通道剪枝策略，并对原策略进行了改进，提高了剪枝率和精度；在这些工作基础上，又衍生出了层剪枝，本身通道剪枝已经大大减小了模型参数和计算量，降低了模型对资源的占用，而层剪枝可以进一步减小了计算量，并大大提高了模型推理速度；通过层剪枝和通道剪枝结合，可以压缩模型的深度和宽度，某种意义上实现了针对不同数据集的小模型搜索。此外，除了剪枝作者进一步在YOLOV3中使用了知识蒸馏策略，进一步压缩模型，具体请看原项目。这是国内的检测大佬维护的开源项目，也可以加入到他们的讨论群去讨论。

YOLO-Lite

论文原文：https://arxiv.org/pdf/1811.05588.pdf
项目地址：https://github.com/reu2018DL/YOLO-LITE
项目特点：论文致力于设计一个网络（cpu速度大于10FPS，PASCAL VOC精度大于30%），因此基于YOLO-v2，提出了一个cpu端的或者端侧的实时检测框架YOLO-LITE。在PASCAL VOC 上取得了33.81%的map，在COCO上取得了12.26%的map，实时性达到了21FPS。
算法原理：作者在论文里面主要证明了两点，一是浅层网络（shallow networks）的对于非GPU快速目标检测应用的能力；二是，证明BN层对于shallow networks是不必要的。
精度和推理时间展示：

效果展示：

总结：精度很低，不过对于网络修改和模型加速有一定的参考价值。

YOLO Nano

论文地址：https://arxiv.org/abs/1910.01271
项目地址：https://github.com/liux0614/yolo_nano
项目特点：来自滑铁卢大学与 Darwin AI 的研究者提出了名为 YOLO Nano 的网络，他们通过人与机器协同设计模型架构大大提升了性能。YOLO Nano 大小只有 4.0MB 左右，比 Tiny YOLOv2 和 Tiny YOLOv3 分别小了 15.1 倍和 8.3 倍，在计算上需要 4.57B 次推断运算，比后两个网络分别少了 34% 和 17%，在性能表现上，在 VOC2007 数据集取得了 69.1% 的 mAP，准确率比后两者分别提升了 12 个点和 10.7 个点。值得一提的是4M是int8量化后的大小。
精度和推理时间展示：

精度和推理时间展示
网络结构：

网络结构
总结：这个开源工程并不是作者公开的，论文中的信息也不足以支撑实现代码的开发以及效果的复现，不过PEP和FCA模块可以作为我们设计网络去借鉴的思路。

附录

YOLOV3 608：http://pjreddie.com/darknet/yolo/
Bag of Freebies for Training Object Detection Neural Networks : https://arxiv.org/abs/1902.04103

作者：BBuf
原文：GiantPandaCV

推荐阅读

更多嵌入式AI技术相关内容请关注嵌入式AI专栏。

前言

推荐项目

Pytorch-YOLOv3

ultralytics-yolov3

YOLOv3-model-pruning

Slim YOLOV3

Anchor聚类 kmeans-anchor-boxes

ASFF

yolov3-channel-and-layer-pruning

YOLO-Lite

YOLO Nano

推荐阅读

目录

【综述】Pytorch YOLO项目推荐 建议收藏学习

前言

推荐项目

Pytorch-YOLOv3

ultralytics-yolov3

YOLOv3-model-pruning

Slim YOLOV3

Anchor聚类 kmeans-anchor-boxes

ASFF

yolov3-channel-and-layer-pruning

YOLO-Lite

YOLO Nano

推荐阅读

目录

【综述】Pytorch YOLO项目推荐建议收藏学习