Визуальный детектор объектов только для задач классификации
Буду очень благодарен, если кто-нибудь поможет мне с моей проблемой.
Я хочу использовать любой детектор объектов (например, YOLO, Faster-RCNN, SSD и т. Д.) На наборе данных COCO только для задачи классификации. Например, если я пропускаю изображение через детектор объектов, и изображение состоит из двух объектов (кошки и собаки). Детектор объектов выдаст мне классификационные баллы для обоих обнаруженных объектов (то есть 80 баллов вероятности для обоих объектов) в конце. (Если кто-то может предоставить код или полезные вещи, я буду очень благодарен).
Я уже использую VGG для задач классификации (которые работают плавно), но мне тоже нужен любой детектор объектов (например, VGG с ImageNet).