You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1.0 KiB

Utility for converting CVAT XML annotation file to YOLO format

Description

Given a CVAT XML, this script reads the CVAT XML and writes the annotations in YOLO format into a given directory. This implementation supports both interpolation tracks from video and annotated images.

Installation

Install necessary packages and create a virtual environment.

sudo apt-get update
sudo apt-get install -y --no-install-recommends python3-pip python3-venv python3-dev
python3 -m venv .env
. .env/bin/activate
cat requirements.txt | xargs -n 1 -L 1 pip install

Usage

Run the script inside the virtual environment:

python converter.py --cvat-xml </path/to/cvat/xml> --image-dir </path/to/images> --output-dir </path/to/output/directory>

Case you need download frames from annotated video file submited to CVAT:

python converter.py --cvat-xml </path/to/cvat/xml> --output-dir </path/to/output/directory> --username <CVAT Username> --password <CVAT Password>

Please run python converter.py --help for more details.