Обучение Yolo на Google Colab завершается с ошибкой в ​​буфере ошибок

Question

Я пытаюсь обучить стандартную неизмененную модель, используя yolov3-tiny.cfg. Я использую Google Colab, потому что не могу позволить себе хороший графический процессор для обучения моих данных. Я тренируюсь на 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, якоря, фильтры = 21 Я попытался с другим файлом cfg, yolov3.cfg без весов и получил ту же ошибку переполнения буфера. Я открыл вопрос о git: https://github.com/pjreddie/darknet/issues/1472

Вот мой файл с колабом: https://colab.research.google.com/drive/1kdwOl9AJ7A7Bn37YcrSVmwGH8U2EI_GC

Я перешел на yolov2.cfg с теми же данными обучения, с той же ошибкой буфера в потоке, поэтому я думаю, что это могут быть данные поезда или файл obj.data.

Answer 1

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

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

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