Как обнаружить только объекты определенной категории при обнаружении тензорного объекта

Question

Записная книжка обнаружения объектов демонстрирует, как модели, предварительно обученные в наборе данных COCO, можно использовать для обнаружения объектов на тестовых изображениях. Тем не менее, модели в блокноте возвращают поля для обнаруженных объектов всех категорий в наборе COCO. Как я могу использовать код для возврата ящиков для объектов только одной категории? Т.е. как я могу получить ящики для предметов, в модели которых уверены, что они, например, люди?

Answer 1

Я только что реализовал решение сам.

проверить def filter_boxes Функция в блокноте https://github.com/fera0013/TrafficLightDetection/blob/master/Traffic-Light-Detection.ipynb, чтобы увидеть, как блоки могут быть отфильтрованы по категориям.