YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone
YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone copied to clipboard
训练Dior遥感数据集,报错shape '[256, 128, 3, 3]' is invalid for input of size 37347
训练命令是: python train.py --data data/dior.data --batch-size 4 --weights weights/yolov3-dior-best.weights -pt --cfg cfg/yolov3/yolov3-onDIOR.cfg --epochs 10 --img-size 608
报错:
Traceback (most recent call last):
File "train.py", line 956, in
把训练命令里pt去掉后,就不会报错。pt的作用是什么尼
-pt是在使用coco预训练模型的时候才使用的,如果你用训练好的模型不需要用-pt