Как получить имя класса напрямую как вывод из yolo tf порции

Question

Я хочу получить вывод из моего пользовательского yolov2 в виде списка словарей идентифицированного объекта.Темный поток имеет функцию return_predict (), которая делает то же самое, но при создании обслуживающего файла.Я не могу предоставить свой выходной слой таким образом, чтобы получить аналогичные результаты.

ссылка на код: https://gist.github.com/sugartom/70b58505bf5f28d1cf5d05904f6c0af2

здесь tfnet.out - дает тензор, но я хочу, чтобы класс был предсказан.

например, результат, который я хочу - [{'label': 'пряжка', 'доверие': 0.32669073, 'topleft': {'x': 368, 'y': 364}, 'bottomright': {'x': 420, 'y': 414}}]

репозиторий github: https://github.com/thtrieu/darkflow

Answer 1

У меня такая же проблема, но пока не могу найти способ.Однако порядок тензора предсказания всегда одинаков.Таким образом, проверяя в начале, результаты могут быть связаны с классами.Для моих выводов, tf порция не поддерживает классификацию.Предсказывать только работы, если не внесены некоторые изменения в источник.