torchsig.image_datasets.datasets.yolo_datasets.YOLODatasetAdapter

class torchsig.image_datasets.datasets.yolo_datasets.YOLODatasetAdapter(dataset: Dataset, class_id: int | None = None)

Bases: Dataset

A class for adapting generic image datasets to YOLO image datasets. Expects a dataset which returns only image tensors, and a class label to apply to the dataset. All returned data will be of the form (image_data, [(class_id, 0.5, 0.5, 1.0 1.0)]), or (image_data, []) if class_id = None

Methods

__init__(dataset: Dataset, class_id: int | None = None)