关于稀疏后剪枝

[rachel-chennn]

请问，我稀疏训练的结果显示 mAP跟正常训练的差不多，如下图

但是放进剪枝读取的时候，就读取到下图这样

mAP变很小，于是我剪枝的结果mAP是零，请问是怎么回事呢

附上我稀疏训练的可视化

谢谢！！！

[ZJU-lishuang]

稀疏度不够吧，剪枝比例超了。稀疏度那里0.5的边界

[rachel-chennn]

稀疏度不够吧，剪枝比例超了。稀疏度那里0.5的边界

那请问大佬，怎样判断稀疏度是不是够了呢，看图嘛？还是看什么指标呢？我后来稀疏成下面图的结果，放进剪枝也还是test the origin model时map就变为很小了，跟上边贴的图结果差不多。 再次感谢！！！

[ZJU-lishuang]

稀疏训练--prune设置的是1吗

[rachel-chennn]

稀疏训练--prune设置的是1吗

参数为--prune 0 --s 0.002 --epoch 350 感谢您的回复！！

[ZJU-lishuang]

我再看看问题吧，prune_yolov5s.py没怎么用，看看和测试函数怎么没搭上

[rachel-chennn]

我再看看问题吧，prune_yolov5s.py没怎么用，看看和测试函数怎么没搭上

非常感谢您，辛苦啦！！

[ZJU-lishuang]

你可以先试试其它的剪枝策略

[rachel-chennn]

你可以先试试其它的剪枝策略

好的！其他剪枝策略有您优先推荐的嘛？我现在就尝试一下

[ZJU-lishuang]

slim_prune_yolov5s_8x.py，其实除prune_yolov5s.py外其它的稀疏策略一致，prune_yolov5s.py实际不实用。

[rachel-chennn]

好的，十分感谢您一直及时的回复~ 我这就去试

[ZJU-lishuang]

我使用项目里的数据集跑prune_yolov5s.py没问题。稀疏模型 last.zip 后缀zip改为pt

[rachel-chennn]

好的感谢！那估计是我的数据集太少了