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

[15050188022]

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

[15050188022]

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

[rachel-chennn]

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

[ZJU-lishuang]

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