Skip to content

Instantly share code, notes, and snippets.

@ikondrat

ikondrat/example-yolo.py

Created February 4, 2025 09:21
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save ikondrat/18bd5765567278c999ea33c2857fd0a7 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save ikondrat/18bd5765567278c999ea33c2857fd0a7 to your computer and use it in GitHub Desktop.
Download ZIP
Example to run lightly-purple on YOLO8 dataset
Raw
example-yolo.py
import os
from lightly_purple import DatasetLoader
# Create a DatasetLoader instance
loader = DatasetLoader()
# Define the path to the dataset (folder containing data.yaml)
dataset_path = os.getenv("DATASET_PATH")
# Load YOLO dataset using data.yaml path
yolo_loader, dataset_id = loader.from_yolo(
f"{dataset_path}/dataset.yaml",
input_split="train",
)
loader.launch()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment