torchsig.image_datasets.datasets.yolo_datasets.YOLOFileDataset

class torchsig.image_datasets.datasets.yolo_datasets.YOLOFileDataset(filepath: str, transforms=None)

Bases: Dataset

A Dataset class for loading image and label files in YOLO format from a root directory Inputs:

filepath: a string file path to a folder containing the yolo dataset transforms: either a single function or list of functions from images to images to be applied to each loaded image; used for adding noise and impairments to data; defaults to None read_black_hot: whether or not to read loaded images as black-hot; this will invert the value of loaded SOIs

Methods

next

__init__(filepath: str, transforms=None)