YOLOv3编译错误（４）：RuntimeError:shape ‘[16, 3, 14, 16, 16]‘ is invalid for input of size 307200
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-tin
·
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)
华为开发者空间，是为全球开发者打造的专属开发空间，汇聚了华为优质开发资源及工具，致力于让每一位开发者拥有一台云主机，基于华为根生态开发、创新。
更多推荐
17
0- 0
已为社区贡献2条内容
所有评论(0)