Сколько (классов) объектов может обнаружить YOLO?

Question

На веб-сайте yolo есть набор весов, предварительно подготовленных для набора данных coco, который состоит из 80 объектов / классов (и я думаю также вес для набора данных voc pascal, который состоит из 20 классов). Есть ли (там) какие-то веса, которые были обучены на наборе данных с большим количеством классов? Я думал, что YOLO может обнаружить тысячи объектов, а не только 80, как кажется. Или я что-то упустил?

Answer 1

Маловероятно, что вы сможете найти веса, обученные на более или менее 80 классах, хотя это вполне возможно. Вы можете восстановить полное репо pjreddie / darknet из GitHub, а затем изменить файл конфигурации, чтобы обучить вашу модель классам «сколько вы хотите».

Полное описание ожидает в файле README.md репозитория.