torchsig.image_datasets.datasets.yolo_datasets.YOLOSOIExtractorDataset¶
- class torchsig.image_datasets.datasets.yolo_datasets.YOLOSOIExtractorDataset(filepath: str, transforms=None, read_black_hot=False, soi_classes: list | None = None, filter_strength=1)[source]¶
Bases:
DatasetA Dataset class for loading marked signals of interest (SOIs) from a yolo format dataset Inputs:
filepath: a string file path to a folder containing images in which all signals of interest have been marked wit ha colored bounding box transforms: either a single function or list of functions from images to images to be applied to each SOI; 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 soi_classes: which classes from the yolo dataset are to be considered signals of interest; None for all classes; defaults to None
Methods