detectionmetrics.datasets package

Submodules

detectionmetrics.datasets.dataset module

class detectionmetrics.datasets.dataset.ImageSegmentationDataset

Bases: object

Parent image segmentation dataset class

append(new_dataset: Self)

Append another dataset with common ontology

Parameters:

new_dataset (Self) – Dataset to be appended

export(outdir: str)

Export dataset dataframe and image files

Parameters:

outdir (str) – Directory where Parquet and image files will be stored

make_fname_global()

Get all relative filenames in dataset and make global

detectionmetrics.datasets.gaia module

class detectionmetrics.datasets.gaia.GaiaImageSegmentationDataset(dataset_fname: str)

Bases: ImageSegmentationDataset

Specific class for GAIA-styled image segmentation datasets

Parameters:

dataset_fname (str) – Parquet dataset filename

detectionmetrics.datasets.goose module

class detectionmetrics.datasets.goose.GooseImageSegmentationDataset(train_dataset_dir: str, val_dataset_dir: str | None = None, test_dataset_dir: str | None = None)

Bases: ImageSegmentationDataset

Specific class for GOOSE-styled image segmentation datasets. All data can be downloaded from the official webpage (https://goose-dataset.de):

Parameters:

  • train_dataset_dir (str) – Directory containing training data

  • val_dataset_dir (str, optional) – Directory containing validation data, defaults to None

  • test_dataset_dir (str, optional) – Directory containing test data, defaults to None

detectionmetrics.datasets.rellis3d module

class detectionmetrics.datasets.rellis3d.Rellis3dImageSegmentationDataset(dataset_dir: str, split_dir: str, ontology_fname: str)

Bases: ImageSegmentationDataset

Specific class for Rellis3D-styled image segmentation datasets. All data can be downloaded from the official repo (https://github.com/unmannedlab/RELLIS-3D):

Parameters:

dataset_dir – Directory where both RGB images and annotations have been

extracted to :type dataset_dir: str :param split_dir: Directory where train, val, and test files (.lst) have been extracted to :type split_dir: str :param ontology_fname: YAML file contained in the ontology compressed directory :type ontology_fname: str

Module contents