как получить имя объекта, обнаруживаемого в тензорном потоке, используя более быструю модель RCNN - PullRequest
0 голосов
/ 16 марта 2019

Я использую API обнаружения объекта Tensflow для обнаружения объекта.Я обучил модель и она работает нормально, но я хочу получить имя обнаруживаемого объекта в виде строки.Может ли кто-нибудь помочь мне с этим?

1 Ответ

0 голосов
/ 16 марта 2019

API предоставляет учебный файл с именем object_detection_tutorial.ipynb , в этом файле функция run_inference_for_single_image возвращает словарь обнаружения output_dict, который содержит ключ detection_classes, это соответствует идентификатору, который вы определили в файле label_map.pbtxt. Также в этом файле переменная category_index содержит карту меток, которая хранится в виде словаря. Таким образом, чтобы просто получить строковое имя для всех обнаруженных объектов, просто добавьте:

string_name = [category_index[i] for i in output_dict['detection_classes']]

после этой строки output_dict = run_inference_for_single_image(image_np, detection_graph) в руководстве.

...