search

SpursLipu / YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone

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

加载yolov3.weights出错 #45

Closed 17868380981 closed 4 years ago

17868380981 commented 4 years ago

Traceback (most recent call last): File "F:/Code/YOLOv3-ModelCompression-MultidatasetTraining-Multibackbone-master/train.py", line 516, in train(hyp) # train normally File "F:/Code/YOLOv3-ModelCompression-MultidatasetTraining-Multibackbone-master/train.py", line 151, in train load_darknet_weights(model, weights, pt=opt.pt) File "F:\Code\YOLOv3-ModelCompression-MultidatasetTraining-Multibackbone-master\models.py", line 566, in load_darknet_weights assert ptr == len(weights) AssertionError

SpursLipu commented 4 years ago

能发一下你使用的指令吗？

17868380981 commented 4 years ago

我只换了cfg，其他都是默认的。注释掉assert ptr == len(weights)就可以正常训练。 另外：双卡训练yolov3-tiny，map是0.65，单卡训练是0.33，数据集dior

SpursLipu commented 4 years ago

换了cfg训练如果类别不一样需要使用-pt指令，不然权重大小是不同的。双卡单卡，不是知道是什么情况。

SpursLipu commented 4 years ago

你用的是什么量化方法