YOLOv3编译错误(4):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=3x(n+5)
版权声明:本文为CSDN博主「weixin_40227656」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_40227656/article/details/115788762
暂无评论