Можем ли мы использовать CPU вместо GPU для обучения привычной модели YOLO для обнаружения объектов

Question

Я хочу обучить модель YOLO для моего набора данных пользовательских объектов. Я читал об этом повсюду на разных сайтах, и все говорят о том, что GPU следует использовать для обучения и запуска пользовательской модели YOLO. Но из-за того, что у меня нет графического процессора, я не понимаю, что делать? Потому что я не могу купить GPU для этого. Кроме того, я читал о Google Colab, но я не могу его использовать, я хочу использовать свою модель в автономной системе. Я испугался, увидев использование системы YOLO, используемой в программе, с github: https://github.com/AhmadYahya97/Fully-Automated-red-light-Violation-Detection.git. Я запускал это на своем ноутбуке с конфигурацией: ОЗУ: 4 ГБ Процессор: Intel i3, 2,40 ГГц ОС: Ubuntu 18.04 LTS

Answer 1

Хотя это будет намного медленнее, да, вы можете использовать ЦП только во время тренировок и прогнозирования. Если вы используете оригинальную Darknet Framework, установите флаг GPU в Makefile при установке darknet на GPU=0.

Как установить darknet: https://pjreddie.com/darknet/install/

Затем вы можете начать тренироваться или прогнозироватьследуя этому руководству: https://pjreddie.com/darknet/yolo/