search

talebolano / yolov3-network-slimming

yolov3 network slimming剪枝的一种实现
345 stars 93 forks source link

mAP #28

Open yyx1107 opened 5 years ago

yyx1107 commented 5 years ago

请问大家有测过剪枝之后的map吗，还是只看得precision和recall呀

liaoyunkun commented 5 years ago

剪枝后的map会下降特别大，从0.4->0.005，只剪枝了20%，有人遇到过吗？

yyx1107 commented 5 years ago

@liaoyunkun 你是按他说的稀疏化训练然后剪枝的吗，你可以再训练一下，提升精度。 我之前在别的上面也剪过一次，边剪边训练map会变好一些，但是没达到之前的精度（你也是做剪枝吗，可以加我交流一下下∩_∩1017176539

PearlDzzz commented 5 years ago

@talebolano @yyx1107 我用自己的数据集 训练yolov3模型, 普通训练100epoch的MAP能达到0.9. 利用sparse_train.py训练yolov3模型, 100epoch的MAP只有0.2. 指令: python sparsity_train.py -sr --s 0.0001 alpha用的 是默认值1.

0-20%:0.000020, 20-40%:0.000123, 40-60%:0.008983, 60-80%: 0.032746, 80-100%:0.540597