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


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

在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,欢迎给大佬加星~

(0)

相关推荐

  • 【目标检测算法解读】yolo系列算法三

    前言 本文是yolo系列算法文章的第三篇,也是目前为止yolo系列算法的最终篇.从原理上看,yolov3并没有实质性的创新,主要是借鉴了一些时下state-of-the-art模型的优秀思想.本文重点 ...

  • 目标检测算法21篇速览:检测网络优化及改进

    作者丨SDC 来源丨GiantPandaCV 编辑丨极市平台 极市导读 本文总结了21篇目标检测算法方面的论文,包括对已有的两种检测网络设计范式的调整和优化,在检测网络中添加注意力模块的方式和方法以及 ...

  • 【AutoML】如何使用强化学习进行模型剪枝?

    大家好,欢迎来到专栏<AutoML>,在这个专栏中我们会讲述AutoML技术在深度学习中的应用,这一期讲述在模型剪枝中的应用. 作者&编辑 | 言有三 我们往期的文章中介绍了各种各 ...

  • 霸榜 GitHub:又一款神器面世!

    不论你需要通用目标检测.实例分割.旋转框检测,还是行人检测.人脸检测.车辆检测等垂类算法: 不论是精度效果超强,还是超轻量适合在边缘部署的算法: 不论你是学术科研工作者,还是产业开发者: 不论你是刚入 ...

  • Micro-YOLO:探索目标检测压缩模型的有效方法(附论文下载)

    重磅干货,第一时间送达 AI算法与图像处理 考研逆袭985,非科班跨行AI,目前从事计算机视觉的工业和商业相关应用的工作.分享最新最前沿的科技,共同分享宝贵的资源资料,这里有机器学习,计算机视觉,Py ...

  • 什么是目标检测中的平均精度均值(mAP)?

    重磅干货,第一时间送达 计算机视觉界已经集中在度量 mAP 上,来比较目标检测系统的性能.在这篇文章中,我们将深入了解平均精度均值 (mAP) 是如何计算的,以及为什么 mAP 已成为目标检测的首选指 ...

  • 基于Mask-GD分割的机器人抓取检测

    重磅干货,第一时间送达 小黑导读 论文是学术研究的精华和未来发展的明灯.小黑决心每天为大家带来经典或者最新论文的解读和分享,旨在帮助各位读者快速了解论文内容.个人能力有限,理解难免出现偏差,建议对文章 ...

  • 广西大学郑含博、李金恒 等:基于改进YOLOv3的电力设备红外目标检测模型

    EVS34圆桌论坛议程:动力电池技术 EVS34圆桌论坛议程:燃料电池商用车发展趋势 EVS34主题论坛议程:智能充换电.充换电基础设施及车网互动 EVS34圆桌论坛议程:下一代电力电子(SiC) E ...

  • 深度学习

    人工智能技术与咨询 4天前 本文来自<系统工程与电子技术>,作者周龙等 摘 要:针对传统雷达图像目标检测方法在海杂波及多种干扰物组成的复杂背景下目标分类识别率低.虚警率高的问题,提出将当前 ...

  • 竞赛冠军方案:2020珠港澳人工智能算法大赛双料冠军解读

    团队介绍 团队来自深圳市威富视界有限公司.中国科学院半导体研究所,队长为宁欣副研究员,成员分别为石园.刘江宽.支金林.王镇.荣倩倩,排名不分先后. 珠港澳人工智能算法赛题介绍 以检测和识别为核心的各项 ...

  • 【目标检测算法解读】yolo系列算法一

    前言 大家好,这是目标检测算法解读系列的第一篇文章,在这个系列中小糖豆会对经典的目标检测算法进行解读,在每个算法原理讲解完后,会有相应的源代码解析.小糖豆认为学习一个算法,既要弄清楚它的原理,又要真正 ...

  • DL之Yolov3:基于深度学习Yolov3算法实现视频目标检测之对《跑男第六季》第四期片视频段进行实时目标检测

    DL之Yolov3:基于深度学习Yolov3算法实现视频目标检测之对<跑男第六季>第四期片视频段进行实时目标检测 完整视频请观看 利用深度学习目标检测之对<跑男第六季>第四期片 ...

  • 揭晓高速公路场景下计算机视觉与交通的秘密

    高速公路场景基于计算机视觉的车辆检测和流量统计 1.文章信息 <Vision-based vehicle detection and counting system using deep lea ...