请教关于剪枝后map为0的问题

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

请教关于剪枝后map为0的问题 #37

Closed mengqingyan97 closed 3 years ago

mengqingyan97 commented 3 years ago

当我使用自己的数据集进行训练时，基础训练过程和稀疏训练过程均正常，但是进行到剪枝的时候，P，R，map，F1的值均变成0；我想请教您这个问题需要如何解决？我的数据集一共2059张图片，做火灾识别，classes=1

ZJU-lishuang commented 3 years ago

降低剪枝率看看剪枝后模型map是否能提升。这个情况一般是模型稀疏度不够导致的。

mengqingyan97 commented 3 years ago

还有一个问题，当我进行剪枝的时候，剪枝前后的map值急剧下降；我稀疏训练得到的map值是0.73，进行到剪枝第一步的时候，let's test the original model first: map直接降到0.25，剪枝后变为0 请问这种情况也是因为稀疏训练不够导致的吗？我基础训练和稀疏训练都是100epoch。

ZJU-lishuang commented 3 years ago

稀疏度不是看map的是看这里记录的东西

mengqingyan97 commented 3 years ago

![Uploading QQ图片20210603192426.png…]()

rachel-chennn commented 2 years ago

您好，我遇到了跟您一样的问题，稀疏后剪枝mAP就变小了，请问是通过重新稀疏训练解决的嘛？