SpursLipu / YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone

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

Assertion error #16

Closed SISTMrL closed 4 years ago

SISTMrL commented 4 years ago

python3 train.py --data data/coco2017.data --batch-size 32 --accumulate 1 -pt --weights weights/yolov3-608.weights --cfg cfg/yolov3/yolov3.cfg --img_size 608 when I train yolov3 with the command above, I encountered the assertion error, due to the python code: assert ptr == len(weights), in line 544, models.py, and I debug the code found that just assign the weights when mdef['type'] == 'convolutional' in line 456, models.py, how I solve this problem? Thanks!

SISTMrL commented 4 years ago

I use the yolov3-608.weights