《模型轻量化-剪枝蒸馏量化系列》YOLOv5无损剪枝（附源码）

今天文章代码不涉密，数据不涉密，使用的是网上开源代码，做了修改，主要介绍如何实现的，另外，数据使用开放数据VisDrone的小部分数据来测试~

今天的文章很短，主要附带一个视频讲解运行过程，我修改的地方就不必说了，代码在文末，可以一键运行。

环境：需要安装：protobuf==3.20.1 其他库见官方yolo所需的环境；

运行顺序：第一步原始训练，得到一个最优mAP等评价指标，记录在小本本上。

第二步：通过调整BN稀疏值（main 参数中的sr），运行train_sparity.py稀疏训练得到一个稍微小一点点的模型（和原始的精度比较，看看哪个稀疏值最好~）

第三步：将上一步的训练好的last.pt 放到prune.py 中进行剪枝，控制剪枝率；剪枝好的模型，在根目录下：pruned_model.pt 是fp32的，你除以2会得到最后的模型大小

第四步：Finetune，用刚刚的pruned模型重新训练，得到最优模型，就是最小且最快，且最好的啦~（和原始和稀疏训练的比较一下哦）

上面是俩个BN的可视化，我们调整稀疏训练里的st值，运行训练&#

《模型轻量化-剪枝蒸馏量化系列》YOLOv5无损剪枝（附源码）相关推荐

arcgis开发多版本之间如何兼容_arcgis api 4.x for js 结合 react 入门开发系列初探篇（附源码下载）...
你还在使用 JQuery 或者 Dojo 框架开发 arcgis api 4.x for js 吗?想试试模块化开发吗?随着前端技术的发展,arcgis api 4.x for js 也有了结合 re ...
python编程选股_随笔吧量化投资选股的python程序实践（附源码）
量化投资选股的python程序实践(附源码) 首先选取沪深股票市场,本人比较关注的12只股票: 000002 万科A,600566 济川药业,300051 三五互联,002039 黔源电力,60087 ...
matlab 突破交易策略,Matlab量化交易策略之横盘突破策略（附源码）
策略名称:横盘突破策略策略说明: 日内交易策略,收盘平仓: 横盘突破在过去30根K线的高低点围绕中轴上下0.5%的范围内波动时: 上轨＝过去30根K线的最高价: 下轨＝过去30根K线的最低价: 当价格 ...
Matlab自适应均线_Matlab量化交易策略之均线打分策略（附源码）
策略思路: 对两条均线的20组参数循环,满足金叉分数+1 当分数大于12分时,多头入场, 当有仓位时,分数小于8分,则平仓出场回测曲线(由Auto-Trader提供回测报告) 均线打分.png (7 ...
python量化选股策略源码_常见的十大量化投资策略（附源码）
量化投资策略是利用量化的方法,进行金融市场的分析.判断和交易的策略.算法的总称.著名的量化投资策略有以下10种(注:策略源码模板不能直接用于实盘交易,仅供探讨交流) 01.海龟交易策略海龟交易策略是 ...
python量化选股策略源码及论文_常见的十大量化投资策略（附源码）
量化投资策略是利用量化的方法,进行金融市场的分析.判断和交易的策略.算法的总称.著名的量化投资策略有以下10种(注:策略源码模板不能直接用于实盘交易,仅供探讨交流) 01.海龟交易策略海龟交易策略是 ...
python量化策略源码_【Python量化投资】趋向系统指标策略 ADX、DMI指标用于股票池（附源码）...
原标题:[Python量化投资]趋向系统指标策略 ADX.DMI指标用于股票池(附源码) [什么是ADX] ADX(average directional indicator) 平均趋向指数,常用的趋 ...
掘金量化 | 短周期量价策略（附源码）
可能不少朋友都有阅读过国泰君安<基于短周期价量特征的多因子选股体系>这篇研报,对其内多达191个量价因子印象深刻.该研报是在2017年中旬发布的,时至今日已过去四年时光,为此大家可能会好奇 ...
【Faster R-CNN论文精度系列】从Faster R-CNN源码中，我们“学习”到了什么？
[Faster R-CNN论文精度系列] (如下为建议阅读顺序) 1[Faster R-CNN论文精度系列]从Faster R-CNN源码中,我们"学习"到了什么? 2[Faste ...
C#进阶系列——一步一步封装自己的HtmlHelper组件：BootstrapHelper（三：附源码）...
阅读目录一.NumberBoxExtensions 二.DateTimeBoxExtensions 1.初始方案 2.改进方案三.TextareExtensions 四.SelectExtensi ...

《模型轻量化-剪枝蒸馏量化系列》YOLOv5无损剪枝（附源码）

《模型轻量化-剪枝蒸馏量化系列》YOLOv5无损剪枝（附源码）相关推荐

最新文章

热门文章