文章目录
- YOLO26 无损剪枝:稀疏训练 + 结构化通道裁剪
- 一、任务
- 二、环境
- 三、流程
- 四、稀疏训练
- 4.1 稀疏正则
- 4.2 BN gamma 分析
- 五、剪枝
- 5.1 通道重要性
- 5.2 结构化剪枝
- 5.3 遍历模型剪枝
- 六、微调
- 七、结果
- 八、消融
- 九、调试
- 十、总结
- 代码链接与详细流程
购买即可解锁1000+YOLO优化文章,并且还有海量深度学习复现项目,价格仅需两杯奶茶的钱,每日更新
YOLO26 无损剪枝:稀疏训练 + 结构化通道裁剪
一、任务
YOLO26s (7.2M 参数, 14.4 MB) ↓ 稀疏训练 (L1 正则化 BN gamma, 30 epoch) ↓ 结构化剪枝 (裁剪 gamma < 阈值 的通道, 50%) ↓ 微调训练 (10 epoch, 恢复精度) ↓ 剪枝后 YOLO26n (3.8 MB, 1.9M 参数)| 阶段 | 模型 | 参数量 | 模型大小 | mAP50 | FPS (GPU) |
|---|---|---|---|---|---|
| 原始 | YOLO26s | 7.2M | 14.4 MB | 0.814 | 168 |
| 稀疏训练 | YOLO26s (sparse) | 7.2 |