marsvision.vision.DeepMarsDataset module

class marsvision.vision.DeepMarsDataset.DeepMarsDataset(root_dir: str)

Bases: torch.utils.data.dataset.Dataset

__init__(root_dir: str)

Initialize transforms. Extract image labels from the dataset in root_dir.

Parameters:

root_dir (str): Root directory of the Deep Mars dataset.