2.1 KiB
Intel OpenVINO toolkit
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/eula.cfgfile.
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.