Скрипт интерпретации для вывода на основе региона Yolo в openvino

Здравствуйте, я новичок в OPENCV / CVAT, я использую openvino для запуска автоматической аннотации, я хочу использовать YoloV3 для этой миссии.

Мне нужно преобразовать модель Yolo в формат OpenVINO для opencv / cvat / auto_annotation. https://github.com/opencv/cvat/tree/develop/cvat/apps/auto_annotation.

Чтобы аннотировать задачу с помощью пользовательской модели, мне нужно подготовить 4 файла:

  1. Конфигурация модели (* .xml) - текстовый файл с конфигурацией сети.
  2. Веса модели (* .bin) - двоичный файл с обученными весами.
  3. Карта меток (* .json) - простой файл json со словарем label_map как объект со строковыми значениями для номеров меток.
  4. Сценарий интерпретации (* .py) - файл, используемый для преобразования выходного сетевого слоя в предопределенную структуру, которая может быть обработана CVAT. Этот код будет запускаться в ограниченной среде Python, но можно использовать некоторые встроенные функции, такие как str, int, float, max, min, range.

Я преобразовал модель Yolo в формат OpenVINO и создал файлы xml и bin. Пишу файл сопоставления lson. Теперь мне нужно написать скрипт Python интерпретации для вывода на основе региона Yolo. Как я могу это сделать? Есть ли файл прерывания от моделей tensorflow к openvino?


person gal leshem    schedule 01.09.2019    source источник


Ответы (2)


Рекомендуем начать с Yolo V3 C ++ или примера Python.

Для C ++:

https://github.com/opencv/open_model_zoo/tree/master/demos/object_detection_demo_yolov3_async

Для образца Python:

https://github.com/opencv/open_model_zoo/tree/master/demos/python_demos/object_detection_demo_yolov3_async

Аналогичное обсуждение можно найти в https://software.intel.com/en-us/forums/computer-vision/topic/816875

person Aswathy - Intel    schedule 24.09.2019

Теперь это связано с CVAT. Сценарий интерпретации и файл json можно найти в репозитории здесь: https://github.com/opencv/cvat/tree/develop/utils/open_model_zoo/yolov3

person Ben Hoff    schedule 09.11.2019