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)
в руководстве.