Quickstart

You will need at least python 3.10 and a recent GPU driver (e.g. NVIDIA 525.147.05).

1. First install the package (best in a python venv).

   python -m pip install "litdet[extras]"
   

2. Train a Faster-RCNN on your dataset annotated in COCO format at /path/to/your/dir/coco_dataset_name:

   light-train task.model=faster-rcnn paths.data_dir=/path/to/your/dir data.data_name=coco_dataset_name task.model.num_classes=10 trainer.max_epochs=100
   

You can find configuration examples in the /examples directory.

To build your own configuration file instead, use our cookiecutter:

cookiecutter https://gitlab.kitware.com/litdet/litdet.git --directory "cookiecutter-litdet"