YOLOv3编译错误(4):RuntimeError:shape ‘[16, 3, 14, 16, 16]‘ is invalid for input of size 307200

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

weixin_40227656

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

单目3D目标检测调研

单目3D目标检测调研 一、 简介 现有的单目3D目标检测方案主要方案主要分为两类,分别为基于图片的方法和基于伪雷达点云的方法。   基于图片的方法一般通过2D-3D之间的几何约束来学习,包括目标形状信息&#xff0