search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
559 stars 137 forks source link

在使用自定义数据集训练时遇到的问题 #38

Open 15050188022 opened 3 years ago

15050188022 commented 3 years ago

你好，感谢您的开源工作。 我在使用你的repo代码对开源数据集NWPUVHR数据集进行剪枝时遇到一些问题，想请教一下您。step1跟step2稀疏训练都没什么问题。 请问第三步中自定义的数据例如fangweisui.data 要怎么生成？我根据coco.data尝试改了一个，但是遇到以下问题： image 请教您我这里的训练命令有没有什么错误。yolov5s_v4.cfg是您提供的，yolov5s_sparsity.pt 是第二步稀疏训练得到的，NWPUVHR.data如下： image

15050188022 commented 3 years ago

data那里是否可以使用原先的.yaml格式文件呢

rachel-chennn commented 2 years ago

请问您问题解决了嘛，遇到了一样的问题

ZJU-lishuang commented 2 years ago

数据格式都是一样的，很容易配置训练文件