查看原文
其他

YOLOv3模型剪枝,瘦身80%,提速100%,精度基本不变

CV君 AI科技大本营 2019-07-10



作者 | CV君

转载自我爱计算机视觉(ID: aicvml)


如果要在实际应用中部署目标检测,你会想到哪项算法?


在52CV目标检测交流群里,被提及最多的,恐怕就是YOLOv3了。


虽然新出的一些算法号称“完胜”“吊打”某某某算法,但YOLOv3 仍是被推荐最多的。


首先它有着目前为止还不错的精度表现,而且计算速度较快,适合在移动端、边缘设备部署。




另一个重要原因是,YOLOv3是被很多人使用验证过的模型,被某个数据库验证过的某算法精度好很重要,但被社区大量验证过效果OK的算法更重要。


今天向大家推荐一个Github新工程,对YOLOv3进行模型剪枝:


https://github.com/Lam1360/YOLOv3-model-pruning


这份代码是基于Python 3.6, PyTorch 实现的。


作者用 YOLOv3 做人手检测(在 oxford hand 数据集上训练的),并进行了模型剪枝,剪枝后YOLOv3 模型的参数量减少 80% ,FLOPs 降低 70%,推断的速度提高了100%,而 mAP 基本保持不变!


可谓是相当成功了!

剪枝前后的对比

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

剪枝前后指标对比:

精度还略微上升了!


作者使用的剪枝算法来自:

Learning Efficient Convolutional Networks Through Network Slimming (ICCV 2017)

https://arxiv.org/abs/1708.06519


剪枝步骤也非常简单:



作者称,后续还会在其他数据集上做实验。


看来剪枝在实际应用部署时还真的很重要呢!


感谢该工程开源者Lam1360,欢迎给大佬加星~


(*本文为 AI科技大本营转载文章,转载请联系原作者)


精彩推荐


“只讲技术,拒绝空谈!”2019 AI开发者大会将于9月6日-7日在北京举行,这一届AI开发者大会有哪些亮点?一线公司的大牛们都在关注什么?AI行业的风向是什么?2019 AI开发者大会,倾听大牛分享,聚焦技术实践,和万千开发者共成长。


目前,大会盲订票限量发售中~扫码购票,领先一步!

    

推荐阅读

你点的每个“在看”,我都认真当成了喜欢

    您可能也对以下帖子感兴趣

    文章有问题?点此查看未经处理的缓存