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

作者 | 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开发者大会，倾听大牛分享，聚焦技术实践，和万千开发者共成长。

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

推荐阅读

大战三回合：XGBoost、LightGBM和Catboost一决高低 | 程序员硬核算法评测
Hinton等人最新研究：大幅提升模型准确率，标签平滑技术到底怎么用?
智能文本信息抽取算法的进阶
入门必备 | 一文读懂神经架构搜索
印度人才出口：一半美国科技企业CEO是印度裔 | 数据分析中印青年
为什么说“大公司的技术顽疾根本挽救不了”
25 年 IT 老兵零基础写小说，作品堪比《三体》| 人物志
中小企业搭建混合云，服务器如何选？
从0到1 | 文本挖掘的传统与深度学习算法
一览微软在机器阅读理解、推荐系统、人机对话等最新研究进展 | ACL 2019
1.2w星！火爆GitHub的Python学习100天刷爆朋友圈！

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

YOLOv3模型剪枝，瘦身80%，提速100%，精度基本不变相关推荐

阿里妈妈搜索广告CTR模型的“瘦身”之路
前言随着号称"地表最强"的千亿参数规模GPT-3模型在 NLP 领域横扫各大数据榜单,大力出奇迹的暴力美学似乎成为了大数据场景建模的不二法门.搜索.推荐和广告场景的 CTR 模型 ...
2篇CIKM详解阿里妈妈搜索广告CTR模型如何低碳瘦身
作为<阿里妈妈搜索广告CTR模型的"瘦身"之路>的姊妹篇,本文将结合团队发表的 CIKM 2021 两篇论文,详解我们在模型瘦身之路上的延续性思考与实践.姊妹篇已经总结 ...
Windows XP瘦身提速大法
<script language='javascript' src='http://www.shiqiaotou.com/Header.js'></script>Windows ...
Windows XP瘦身提速大法！
Windows XP系统占用的空间很大,但是有很多文件并不是正常运行系统所必须的,而且也不是每个用户都需要的,因此我们可以通过清理系统中不用的文件来进行Windows XP的瘦身和提速. 一.Dcum ...
如何让你的YOLOV3模型更小更快？
1. 前言之前讲过关于模型剪枝的文章深度学习算法优化系列七 | ICCV 2017的一篇模型剪枝论文,也是2019年众多开源剪枝项目的理论基础并分析过如何利用这个通道剪枝算法对常见的分类模型如VG ...
极智AI | 谈谈模型剪枝之通道剪枝策略
欢迎关注我的公众号 [极智视界],获取我的更多笔记分享大家好,我是极智视界.本文主要聊一聊模型剪枝之通道剪枝相关实践分享. 模型剪枝是模型推理加速的重要实现技术手段,模型剪枝往往期望达到 ...
【AI周报】谷歌、Meta如何给大模型瘦身？；Meta开放1750亿参数大模型，媲美GPT-3
01 行业大事件谷歌.Meta如何给大模型瘦身?谷歌工程师亲述部署秘籍(免费书) 这是一本正在写作的新书,目前前四章已经放出了草稿. 过去几年,GPT-3 等几个大模型展示了大数据.大算力的力量,它 ...
【剪枝算法】通过网络瘦身学习高效的卷积网络Learning Efficient Convolutional Networks through Network Slimming论文翻译
此论文翻译仅仅用于自己方便学习.译文来自其他处. 在许多实际应用中部署深度卷积神经网络(CNN)很大程度上受到其计算成本高的限制.在本文中,我们提出了一种新的CNNs学习方案,能同时1)减小模型大小; ...
Windows XP 瘦身提速优化技巧大全
1. 缩短等待时间打开注册表编辑器,找到 HKEY_LOCAL_MACHINE＼System＼CurrentControlSet＼Control,将 WaitToKillServiceTimeout ...

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

剪枝前后的对比

大战三回合：XGBoost、LightGBM和Catboost一决高低 | 程序员硬核算法评测

Hinton等人最新研究：大幅提升模型准确率，标签平滑技术到底怎么用?

智能文本信息抽取算法的进阶

入门必备 | 一文读懂神经架构搜索

印度人才出口：一半美国科技企业CEO是印度裔 | 数据分析中印青年

为什么说“大公司的技术顽疾根本挽救不了”

25 年 IT 老兵零基础写小说，作品堪比《三体》| 人物志

中小企业搭建混合云，服务器如何选？

从0到1 | 文本挖掘的传统与深度学习算法

一览微软在机器阅读理解、推荐系统、人机对话等最新研究进展 | ACL 2019

1.2w星！火爆GitHub的Python学习100天刷爆朋友圈！

YOLOv3模型剪枝，瘦身80%，提速100%，精度基本不变相关推荐

最新文章

热门文章