SpursLipu / YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone

YOLO ModelCompression MultidatasetTraining
GNU General Public License v3.0
444 stars 136 forks source link

稀疏训练准确率太低 #98

Closed chenxyyy closed 3 years ago

chenxyyy commented 3 years ago

@SpursLipu 你好,感谢你的工作。

我在使用您的工程对yolov3剪枝的时候,在第二步稀疏化训练coco2017数据集,使用的是您所提供的预训练权重,但是mAP下降的太多了。之前是0.53左右,稀疏化训练后只有0.32.我目前训练到20个epoch了,是训练的时间太短了吗?

我是用的命令是:

python train.py --data data/coco2017.data --batch-size 22 -pt --weights weights/yolov3-608.weights --cfg cfg/yolov3/yolov3.cfg -sr --s 0.001 --prune 0
SpursLipu commented 3 years ago

coco的话确实可能是训练时间太短,你可能要训练得久一点