Как я могу удалить некоторые объекты из набора данных COCO для YOLOv3

Question

Я следовал Обучаем пользовательские данные , чтобы обучить мою пользовательскую модель. Но есть некоторые классы / объекты в наборе данных COCO, которые также хотят включить. Есть ли быстрый способ извлечь только те классы из набора данных. Я думаю написать сценарий для этого, но мне кажется, что должен быть мой гораздо более быстрый способ выполнить sh это.

Answer 1

Вы можете либо изменить метки, либо модифицировать загрузчик данных для фильтрации по классу. Загрузчик данных находится здесь:

https://github.com/ultralytics/yolov3/blob/master/utils/datasets.py#L258

Последний вариант - тренироваться на всех 80 классах или просто использовать предварительно подготовленные веса, а затем запускать обнаружение только для некоторые классы:

python3 detect.py --classes 0,3,5

На это ответил Гленн Джохер в GitHub. Оригинальные номера ссылка