Google Colab上的Yolo培训因错误缓冲区溢出而崩溃

时间:2019-03-06 20:00:39

标签: buffer-overflow google-colaboratory training-data yolo

我正在尝试使用yolov3-tiny.cfg训练标准的未修改模型。我使用Google Colab是因为我买不起一个好的GPU来训练我的数据。我正在2节课上训练。并发生此错误。

我传递给Colab的命令:

Sub Test()
    Sheets("CJI Data").Select
    Dim LastRow As Long
    Dim Sht As Worksheet
    Set Sht = ActiveWorkbook.Sheets("CJI Data")
    LastRow = Sht.Range("X1048576").End(xlUp).Row
    Range("Y3:AB3" & LastRow).FillDown
    Range("A2:AB2").Select
End Sub

模型加载后,我看到以下输出: ” *检测到缓冲区溢出* :./darknet已终止

我在yolov3-tiny.cfg中修改的全部是class = 2,anchors,filters = 21 我尝试使用其他cfg文件yolov3.cfg,但没有权重,并且得到了相同的缓冲区溢出错误。 我在git上打开了一个问题:https://github.com/pjreddie/darknet/issues/1472

这是我的合作文件: https://colab.research.google.com/drive/1kdwOl9AJ7A7Bn37YcrSVmwGH8U2EI_GC

我更改为yolov2.cfg,使用了相同的训练数据,相同的缓冲区溢出错误,所以我认为可能是训练数据或obj.data文件。

1 个答案:

答案 0 :(得分:0)

所有有关文本文件格式的内容。 文本文件是在 Windows操作系统上生成的,并且Google Colab运行了 Ubuntu VM计算机,因此我用line.rstrip('\r')格式化了.txt。

我遇到了一些其他错误,例如无法加载.jpg文件,它需要来自darknet目录的相对路径,因此在train.txt中,路径:/data/obj/1.jpg还不够{{ 1}}。

为此,我可以使用Google Colab在我的个人数据集上训练yolo。