search

coldlarry / YOLOv3-complete-pruning

提供对YOLOv3及Tiny的多种剪枝版本，以适应不同的需求。
805 stars 214 forks source link

我使用和您相同的数据集以及参数训练，channel prun结果相仿，layer prun后的map微调前只有20，可能是什么问题呢？？ #35

Closed px2017 closed 4 years ago

coldlarry commented 4 years ago

您好 如果是和我相同的数据集，可能是稀疏化的问题，还有可能是您代码clone的比较早？后面我对代码进行了简单修改，不知道您更新了没有。

px2017 commented 4 years ago

谢谢您的回复！ 代码是我这周二12月3号clone下来的，之后您应该没有修改代码。所以应该不是代码版本的问题。 这是稀疏化的训练结果，第一张图像是map，第二张是gamma值，看起来稀疏化训练没有什么问题，第三张是剪枝结果。 参数和您的设置都一样，唯一不同的是我没有用spp，就用的普通的yolov3，是这个导致layer prun效果差吗？

python layer_channel_prune.py --data data/oxfordhand.data --cfg cfg/yolov3-hand.cfg --weights weights/last.pt --global_percent 0.8 --layer_keep 0.01 --shortcuts 16

python layer_channel_prune.py --cfg cfg/yolov3-spp-hand.cfg --data data/oxfordhand.data --weights weights/last.pt --global_percent 0.93 --layer_keep 0.01 --shortcuts 16

------------------ 原始邮件 ------------------ 发件人: "coldlarry"<notifications@github.com>; 发送时间: 2019年12月6日(星期五) 中午12:39 收件人: "coldlarry/YOLOv3-complete-pruning"<YOLOv3-complete-pruning@noreply.github.com>; 抄送: "彭响"<253358407@qq.com>;"State change"<state_change@noreply.github.com>; 主题: Re: [coldlarry/YOLOv3-complete-pruning] 我使用和您相同的数据集以及参数训练，channel prun结果相仿，layer prun后的map微调前只有20，可能是什么问题呢？？ (#35)

您好 如果是和我相同的数据集，可能是稀疏化的问题，还有可能是您代码clone的比较早？后面我对代码进行了简单修改，不知道您更新了没有。

— You are receiving this because you modified the open/close state. Reply to this email directly, view it on GitHub, or unsubscribe.