YOLOv5剪枝 | 模型剪枝实战篇 | 2/2

YOLOv5剪枝 | 模型剪枝实战篇 | 2/2_第1张图片

本篇博文所用代码为开源项目修改得到,且不适合基础太差的同学。

本篇文章主要讲解代码的使用方式,手把手带你实现YOLOv5模型剪枝操作。

文章目录

      • 0. 环境准备
      • 1. 使用YOLOv5训练自己的模型
      • 2. 对训练好的模型进行稀疏训练
      • 3. 对稀疏训练后的模型进行剪枝
      • 4. 对剪枝后的网络模型微调
      • 5. 测试微调后的模型
      • 6. 总结
      • 7. 源码
      • 8. 参考代码

0. 环境准备

终端键入:

pip install -r requirements.txt -i https

你可能感兴趣的:(YOLOv5进阶实战,剪枝,深度学习,计算机视觉)