search

Lam1360 / YOLOv3-model-pruning

在 oxford hand 数据集上对 YOLOv3 做模型剪枝（network slimming）
MIT License
1.67k stars 431 forks source link

mAP很低 #72

Open jenn0727 opened 4 years ago

jenn0727 commented 4 years ago

训练自己的数据集，数据集小，4k多张图，共20类，不管是训练baseline还是稀疏化，mAP怎么都很低，学习率调整了也上不去。 1、在darknet的c的源码上训练baseline，用 eriklindernoren 的test测mAP是0.73，用本项目test测是0.51，并且调整batch size会对结果产生影响，不明原因中。。。 2、用本项目训练baseline，调整了很多次mAP都只能达到0.3 3、用本项目训练稀疏化，目前300epoch也只达到0.2 4、用 eriklindernoren 训练baseline ：7月的时候做的 mAP大概也只有0.54 （eriklindernoren 的test.py） 现在重新训练尝试中

mi696 commented 3 years ago

你好，我也想训练自己的数据集，但我的数据集标签是xml格式，转换为mat文件后会出现如下的报错boxes = content["boxes"] KeyError: 'boxes'，想问请教一下你自己的数据集是怎么设置格式的？