求prune_utils.py - Githubissues

Lam1360 / YOLOv3-model-pruning

在 oxford hand 数据集上对 YOLOv3 做模型剪枝（network slimming）

MIT License

1.67k stars 431 forks source link

Open liaoyunkun opened 5 years ago

violet17 commented 5 years ago

估计只有1000星的时候，才会开源了。。。

violet17 commented 5 years ago

所以我自己补了一个。。

liaoyunkun commented 5 years ago

@violet17 请问你有在oxfordhand上面重现作者的结果吗？我这边根本训练不到MAP为0.7的水平

violet17 commented 5 years ago

@liaoyunkun 我在别的数据集上。。。一个简单的数据集，剪掉10%的channel，性能都下降了不少

violet17 commented 5 years ago

@liaoyunkun 你有微信么私聊啊

liaoyunkun commented 5 years ago

@violet17 我稀疏化训练得到一个MAP为0.43左右的模型[80%的γ接近0]，剪去80%的channel得到一个MAP接近为0，Finetune一个Epoch后得到一个MAP为0.41·的模型，很迷

smartcai commented 5 years ago

repo主这么久都没开源唉

1451595897 commented 5 years ago

@violet17 你好，你补充了prune_utils.py文件后，有没有训练过自己的数据集啊？效果怎么样？能复现原作者的效果吗？

violet17 commented 5 years ago

@1451595897 还在训练。我没用那个手的数据集，我用的是我自己的数据集。

mi696 commented 3 years ago

@violet17 你好，你自己训练的数据集是voc格式的吗？我自己的数据集标签是xml的，转化为mat文件后运行会报错 boxes = content["boxes"] KeyError: 'boxes'，请教一下你自己的数据集是怎么设置格式的？