Train a YOLO model for object detection with oriented bounding boxes¶

This notebook trains a YOLO model for object detection with oriented bounding boxes on the HRSC2016-MS dataset.

The original dataset can be found here.

We re-use the Tables created in create-custom-obb-table.ipynb.

Project setup¶

[ ]:

PROJECT_NAME = "3LC Tutorials - OBBs"
DATASET_NAME = "HRSC2016-MS"
DOWNLOAD_PATH = ""

Imports¶

[ ]:

%pip install 3lc-ultralytics

[ ]:

import tlc
from tlc_ultralytics import YOLO, Settings

Load tables¶

[ ]:

train_table = tlc.Table.from_names("train", DATASET_NAME, PROJECT_NAME)
val_table = tlc.Table.from_names("val", DATASET_NAME, PROJECT_NAME)

Train a model¶

[ ]:

model = YOLO("yolo11n-obb.pt")
settings = Settings(
    project_name=PROJECT_NAME,
    run_name="train-yolon-obb-hrsc2016-ms",
    image_embeddings_dim=2,
)

model.train(
    tables={"train": train_table, "val": val_table},
    settings=settings,
    label_column_name="obb",
    epochs=10,
    workers=0,
)