search

PINTO0309 / OpenVINO-YoloV3

YoloV3/tiny-YoloV3+RaspberryPi3/Ubuntu LaptopPC+NCS/NCS2+USB Camera+Python+OpenVINO
https://qiita.com/PINTO
Apache License 2.0
538 stars 167 forks source link

Convert Yolo model to OpenVINO format for opencv/cvat/auto_annotation task #53

Open alontrais opened 4 years ago

alontrais commented 4 years ago

I need to convert Yolo model to OpenVINO format for opencv/cvat/auto_annotation. I use opencv CVAT for auto annotation https://github.com/opencv/cvat/tree/develop/cvat/apps/auto_annotation.

To annotate a task with a custom model I need to prepare 4 files:

  1. Model config (*.xml) - a text file with network configuration.
  2. Model weights (*.bin) - a binary file with trained weights.
  3. Label map (*.json) - a simple json file with label_map dictionary like object with string values for label numbers.
  4. Interpretation script (*.py) - a file used to convert net output layer to a predefined structure which can be processed by CVAT. This code will be run inside a restricted python's environment, but it's possible to use some builtin functions like str, int, float, max, min, range.

How can I get these files?

benhoff commented 4 years ago

https://github.com/opencv/cvat/pull/794