search

ZJU-lishuang / yolov5_prune

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

感谢大佬为开源作出的贡献。 #116

Open Derbins opened 1 year ago

Derbins commented 1 year ago

我通过正常训练和稀疏训练自己的数据集得出来的权重文件，在进行step3的时候发现一个问题，您的工作是在hand单类数据集上进行的，在step3中所使用的参数--data oxhand.data与u版的参数--data xxx.yaml 有所不同，并且Test.py文件对数据集的处理都是按照单类处理的，我数据集现在有5类，我想问您的问题是我该如何在自己的数据集下对训练参数或者源码作出修改呢？

ZJU-lishuang commented 1 year ago

修改配置文件cfg中的类别数吧，具体可以边debug边修改 这代码跑过多个模型，不只是适配单类别的

ZJU-lishuang commented 1 year ago

不建议零基础直接来剪枝