Yolov5模型压缩-slim network

本文介绍了如何在YOLOv5模型上应用网络瘦身进行模型压缩。通过学习有效卷积网络,利用BN层的缩放因子γ评估通道重要性,并设置阈值裁剪不重要的通道。经过正常训练、稀疏化训练、模型剪枝和微调训练四个步骤,最终实现模型的轻量化。在剪枝过程中,通过调整稀疏化因子并观察权重变化,以确保模型精度。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

参考论文:Learning Efficient Convolutional Networks through Network Slimming(https://arxiv.org/pdf/1708.06519.pdf)
基本原理:利用BN层中的缩放因子γ来控制通道的重要性,设置合适的阈值减掉不那么重要的通道,从而实现网络的轻量化,一般为了恢复模型的精度,会在剪枝训练之后进行微调训练。这个过程可以重复多次,直到获得满意的模型。
在这里插入图片描述在yolov5上进行剪枝训练,流程如下:

1、正常训练
和yolov5的正常训练一样
2、稀疏化训练
新建train_sparse.py,复制train.py的代码,需要改动反向传播、优化器,添加稀疏化因子和稀疏化可视化

            # Backward
            # scaler.scale(loss).backward()
  
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值