YOLOv3编译错误（４）：RuntimeError:shape '[16, 3, 14, 16, 16]' is invalid for input of size 307200

系统环境：Ubuntu18.04＋u版yolov3＋VOC2007数据集

执行命令： python train.py --data data/voc2007.data --weights weights/yolov3-tiny.conv.15 --cfg cfg/yolov3-tiny.cfg --epochs 10 --device 1

错误提示：RuntimeError:shape '[16, 3, 14, 16, 16]' is invalid for input of size 307200

问题分析：维度不匹配，即：修改的.cfg文件yolo层前的classes与filters与自定义的数据集不匹配。

解决方法：修改classes为类别数，filters=3ｘ(n+5)