前言
问题的出现场景:最近在学习YOLO4 , 在用YOLO4跑自己的数据集的时候, 发现会爆出
训练集Layer before convolutional layer must output image.: No error的错误。
提示:以下是本篇文章正文内容,下面案例可供参考
一、解决办法
检查自定义的cfg文件, 输入图像的大小设置如下
height和width 都设置成32 的倍数的就不会爆出这个问题了。 我这里设置为416 、 416
二、后续问题
注意将 大小设置成自己数据集中图片的大小 ,否则可能无法打开图片,报错如下
Can’t load image xxxxxxxxxxxxxxxxxx
总结
如果有帮助的话 点个赞再走吧!