search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
559 stars 137 forks source link

减枝后map不一致 #44

Open HouBiaoLiu opened 3 years ago

HouBiaoLiu commented 3 years ago

image

两者mAP不一致 正常吗

ZJU-lishuang commented 3 years ago

稀疏训练没到位就会这样，模型本身效果也不佳

HouBiaoLiu commented 3 years ago

这两者计算mAP的方式是怎样的？ 感觉训练还可以啊 我稀疏训练 学习率lr0: 0.001 lrf: 0.02 image 取150轮做裁剪

ZJU-lishuang commented 3 years ago

稀疏训练不是看map的，代码使用tensorboard记录了参与稀疏的bn层的Gmma权重变化，正常训练时Gmma总体上分布在1附近类似正态分布，稀疏过程Gmma大部分逐渐被压到接近0，接近0的通道其输出值近似于常量，可以将其剪掉。

HouBiaoLiu commented 3 years ago

image 那我该选哪轮保存的模型做裁剪呢

ZJU-lishuang commented 3 years ago

yolov5由于mosaic数据增强的存在，可以先不考虑过拟合，选最后的模型