API обнаружения объектов Tensorflow MS-COCO Оценка нескольких категорий Сбор результатов
Я использую API-интерфейс обнаружения тензорного потока с оценкой кокоса
Я сделал пользовательский набор данных о дорогах, и у меня есть две категории: "автомобиль" и "тяжелый автомобиль".
Теперь результаты показывают, как это
введите описание изображения здесь
То, что я хочу показать, это результаты оценки каждой категории: автомобиль и тяжелый автомобиль
Сначала я подумал, что файлы 'coco_tools.py' и 'coco_evaluation.py' будут работать. Они находятся в папке "models / research / object_detection / metrics".
Есть некоторые строки, которые, кажется, влияют на результаты оценки. Например, "include_metrics_per_category" или "all_metrics_per_category".
Однако, похоже, это связано с оригинальными категориями метрики COCO. Мой набор данных имеет свою собственную категорию: автомобиль и тяжелая техника.
Если я изменю 'include_metrics_per_category' и 'all_metrics_per_category' в 'coco_tools.py' и 'coco_evaluation.py' на True, ошибка будет отображаться, как показано ниже
введите описание изображения здесь
Фактическая ошибка намного длиннее, но в основном она говорит об одном и том же. "Статистика категории не существует" - ключевое слово.
Как я могу получить результат оценки для каждой категории? и если это возможно, как я могу сделать текстовый файл, который включает в себя эти результаты метрики.
Спасибо!