google colab 上的 Yolo 训练因错误缓冲区溢出而崩溃

Question

我正在尝试使用 yolov3-tiny.cfg 训练标准的未修改模型。我正在使用 Google 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 中修改的全部是类 = 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 文件。

Answer 1

一切都是关于文本文件的格式。

文本文件是在 Windows OS 上生成的，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。

google colab 上的 Yolo 训练因错误缓冲区溢出而崩溃

Yolo training on google colab crashes with error buffer over flow

buffer-overflow

google-colaboratory

training-data

yolo