Welcome to Teklia YOLO

This repository allows to handle YOLO and to extract data from Arkindex in the correct format for training and/or evaluating a YOLO model. It is based on Ultralytics' implementation of YOLO and therefore prepares the data in the corresponding format.

YOLO models

Using the Ultralytics library, it is possible to train models for 4 for different tasks:

  • Classification

  • Detection: Bounding boxes / Oriented bounding boxes

  • Segmentation

Data extraction and formatting

This repository allows users to extract and prepare data in the right format for training a YOLO model. More details in the dedicated page.

Training and Evaluating

This repository also allows users to train and evaluate YOLO models. The code is based on the Ultralytics' implementation. More details in the dedicated page.