Как отфильтровать классы из набора данных COCO?

Question

Я бы хотел тренировать модель Yolo с набором данных COCO. Поскольку существует более 80 классов, как я могу отфильтровать его? Мне просто нужен человек классов и машина.

Answer 1

Для простого и простого способа выполните следующие действия:

• Изменить (или скопировать для резервного копирования) файл coco.names в darknet\data\coco.names
• Удалить все другие классы, кроме человека и автомобиля
• Измените свой файл cfg (например, yolov3.cfg), измените 3 класса в строке 610, 696, 783 с 80 на 2
• Измените 3 фильтра в файле cfg в строке 603, 689, 776 с 255 на (классы + 5) x3 = 21
• Запустить детектор ./darknet detector test cfg/coco.data cfg/yolov3.cfg yolov3.weights data/person.jpg

Для более продвинутого пути вы можете использовать это репо для создания наборов данных yolo на основе вока, кокоса или открытых изображений. https://github.com/holger-prause/yolo_utils

См. Также: Как загрузить определенную часть набора данных Coco?