size mismatch for yolo_head_P5.bias: copying a param with shape torch.Size([255]) from checkpoint, the shape in current model is torch.Size([75]).
错误:
预测图片时,出现错误
错误原因:
yolo.py中,
“classes_path”: ‘model_data/voc_classes.txt’
分类数量不匹配
改为:
“classes_path”: ‘model_data/coco_classes.txt’
问题解决