iaa_od.models.ground_truth_protocol

Classes

GroundTruthProtocol(*args, **kwargs)

Protocol representing the abstract structure of a Ground Truth.
class iaa_od.models.ground_truth_protocol.GroundTruthProtocol(*args, **kwargs)[source]

Bases: Protocol

Protocol representing the abstract structure of a Ground Truth. This protocol defines the structure of GroundTruth and RandomGroundTruth.

name

Name of the 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 annotations.

Type:

dict[str, list[AnnotationProtocol]]

categories_dict

A dictionary mapping category IDs to category names.

Type:

dict[int, str]

annotations: dict[str, list[AnnotationProtocol]]
categories_dict: dict[int, str]
images: dict[str, Image]
name: str