我正在尝试使用yolov3-tiny.cfg训练标准的未修改模型。我正在使用谷歌Colab,因为我买不起一个好的GPU来训练我的数据。我正在训练2节课。并且发生了这个错误..
我传给Colab的命令:
!./darknet detector train build/darknet/x64/data/obj.data build/darknet/x64/cfg/yolov3-tiny.cfg yolov3-tiny.weights -dont_show
模型加载后,我看到这个输出:“*检测到缓冲区溢出*:./ darknet终止
我在yolov3-tiny.cfg中修改的所有类都是classes = 2,anchors,filters = 21我尝试使用不同的cfg文件,没有权重的yolov3.cfg并且得到了相同的缓冲区溢出错误。我在git:https://github.com/pjreddie/darknet/issues/1472上打开了一个问题
这是我的colab文件:https://colab.research.google.com/drive/1kdwOl9AJ7A7Bn37YcrSVmwGH8U2EI_GC
我使用相同的训练数据更改为yolov2.cfg,相同的缓冲区溢出错误,所以我认为它可能是列车数据或obj.data文件。
这完全是关于文本文件的格式化。文本文件是在Windows操作系统上生成的,而Google Colab运行的是Ubuntu VM机器,因此我使用line.rstrip('\r')
格式化.txt。
我有一些额外的错误,如无法加载.jpg文件,它需要来自darknet目录的相对路径,所以在train.txt路径:/data/obj/1.jpg
它是不够的它需要/build/darknet/x64/data/obj/1.jpg
。
目前我可以在我的个人数据集上使用Google Colab训练yolo。