Skip to content

Latest commit

 

History

History
 
 

openvino

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

Requirements

  • Intel Core with 6th generation and higher or Intel Xeon CPUs.

Preparation

  • Download the latest OpenVINO toolkit .tgz installer (offline or online) for Ubuntu platforms. Note that OpenVINO does not maintain forward compatability between Intermediate Representations (IRs), so the version of OpenVINO in CVAT and the version used to translate the models needs to be the same.
  • Put downloaded file into cvat/components/openvino.
  • Accept EULA in the cvat/components/openvino/eula.cfg file.

Build docker image

# From project root directory
docker-compose -f docker-compose.yml -f components/openvino/docker-compose.openvino.yml build

Run docker container

# From project root directory
docker-compose -f docker-compose.yml -f components/openvino/docker-compose.openvino.yml up -d

You should be able to login and see the web interface for CVAT now, complete with the new "Model Manager" button.

OpenVINO Models

Clone the Open Model Zoo. $ git clone https://github.com/opencv/open_model_zoo.git

Install the appropriate libraries. Currently that command would be $ pip install -r open_model_zoo/tools/downloader/requirements.in

Download the models using downloader.py file in open_model_zoo/tools/downloader/. The --name command can be used to specify specific models. The --print_all command can print all the available models. Specific models that are already integrated into Cvat can be found here.

From the web user interface in CVAT, upload the models using the model manager. You'll need to include the xml and bin file from the model downloader. You'll need to include the python and JSON files from scratch or by using the ones in the CVAT libary. See here for instructions for creating custom python and JSON files.