谷歌colab上的Yolo训练崩溃与错误缓冲区流量

问题描述 投票:0回答:1

我正在尝试使用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文件。

buffer-overflow google-colaboratory training-data yolo
1个回答
0
投票

这完全是关于文本文件的格式化。文本文件是在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。

© www.soinside.com 2019 - 2024. All rights reserved.