A package and command-line tool to quickly get from raw data to a production model. The main target is to train an object detector and a image classfier / similarity search model to detect and recognize fine-grained objects on images and video.
This package is built upon an open-source computer vision data management tool FiftyOne. It also provides an integration with a data annotation tool CVAT and a model-training framework Lightning Flash that are used to prepare data and train models.
Docs are temporarily hosted on Yandex.Cloud.
The package is structured as follows:
- data: data prep for training
- models: train, evaluate and export models
- services: interact with external services
This package is managed with Poetry. To install the package in development mode, run the following command:
poetry install
pre-commit install