Yolo Training на google colab вылетает из-за переполнения буфера ошибок

Я пытаюсь обучить стандартную немодифицированную модель с помощью yolov3-tiny.cfg. Я использую Google Colab, потому что не могу позволить себе хороший графический процессор для обучения моих данных. Тренируюсь на 2-х классах. и возникла эта ошибка ..

Команда, которую я передаю Колабу:

!./darknet detector train build/darknet/x64/data/obj.data build/darknet/x64/cfg/yolov3-tiny.cfg yolov3-tiny.weights -dont_show

После загрузки модели я вижу следующий вывод: «* обнаружено переполнение буфера *: ./darknet terminated

Все, что я изменил в yolov3-tiny.cfg, - это классы = 2, привязки, фильтры = 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.


training-data yolo google-colaboratory buffer-overflow
person Cristi.Fatu    schedule 06.03.2019    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Все дело в форматировании текстовых файлов.

Текстовые файлы были созданы в ОС Windows, а Google Colab работает на виртуальной машине Ubuntu, поэтому я отформатировал .txt с помощью line.rstrip('\r').

У меня возникли дополнительные ошибки, например, не могу загрузить файл .jpg, ему нужен относительный путь из каталога darknet, поэтому в train.txt путь: /data/obj/1.jpg недостаточно, требуется /build/darknet/x64/data/obj/1.jpg.

На данный момент я могу тренировать йоло с помощью Google Colab на моем наборе личных данных.

person Cristi.Fatu    schedule 07.03.2019