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

文章目录

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

1. 准备自己的数据集

1.1 下载项目文件

  • 准备好备注的数据集进行训练,我这里给出了标注好的足球的数据集。从百度网盘下载到项目目录下并解压,网盘地址见文末
    • VOCdevkit_ball.zip
    • testfiles.zip
    • prepare_data.py

1.2 解压建立或自行建立数据集

  • 使用PASCAL VOC数据集的目录结构, 建立文件夹层次为 <

你可能感兴趣的:(模型轻量化,剪枝,深度学习)