iaa_od.models.random_ground_truth

Classes

RandomGroundTruth(*[, deviation])

A class to represent a randomly generated ground truth dataset, based on statistics sampled from existing ground truths.

class iaa_od.models.random_ground_truth.RandomGroundTruth(*, deviation=0.5, gts)[source]

Bases: object

A class to represent a randomly generated ground truth dataset, based on statistics sampled from existing ground truths.

name

The name of the randomly generated ground truth dataset.

Type:

str

images

A dictionary mapping image filenames to Image objects.

Type:

dict[str, Image]

annotations

A dictionary mapping image filenames to lists of RandomAnnotation objects.

Type:

dict[str, list[RandomAnnotation]]

categories_dict

A dictionary mapping category IDs to category names.

Type:

dict[int, str]

deviation

The standard deviation to use for the Gaussian distribution when generating random bounding box sizes. Default is 0.5.

Type:

float

annotations: dict[str, list[RandomAnnotation]]
categories_dict: dict[int, str]
deviation: float
gts: InitVar
images: dict[str, Image]
name: str