- Intel Core with 6th generation and higher or Intel Xeon CPUs.
- 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.
# From project root directory
docker-compose -f docker-compose.yml -f components/openvino/docker-compose.openvino.yml build
# 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.
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.