yolov5剪枝实战4: 正常训练和稀疏化训练

文章目录

• • 1. 准备自己的数据集
  • • 1.1 下载项目文件
    • 1.2 解压建立或自行建立数据集
  • 1.3 生成训练集和验证集文件
  • 2. 修改配置文件
  • • 2.1 新建文件data/voc_ball.yaml
    • 2.2 新建文件models/yolov5s_ball.yaml
  • 3. 训练自己的数据集
  • 4. 代码讲解
  • • 4.1 修改BN的梯度
    • 4.2 BN权重可视化
  • 4. 测试训练出的网络模型
  • 5. 稀疏化训练
  • 6 代码下载