pysolotools.core.iterators package

Submodules

pysolotools.core.iterators.frame_iterator module

class pysolotools.core.iterators.frame_iterator.FramesIterator(data_path: str, metadata: DatasetMetadata, start: int = 0, end: Optional[int] = None)

Bases: object

SENSORS = [{'sensor': 'type.unity.com/unity.solo.RGBCamera', 'annotations': [<class 'pysolotools.core.models.solo.BoundingBox2DAnnotation'>, <class 'pysolotools.core.models.solo.BoundingBox3DAnnotation'>, <class 'pysolotools.core.models.solo.InstanceSegmentationAnnotation'>, <class 'pysolotools.core.models.solo.SemanticSegmentationAnnotation'>]}]

parse_frame(f_path: str) → Frame

Parses a json file to a pysolo Frame model.

Parameters: f_path (str) – Path to a step in a sequence for a frame.
Return type: Frame

Module contents

class pysolotools.core.iterators.FramesIterator(data_path: str, metadata: DatasetMetadata, start: int = 0, end: Optional[int] = None)

Bases: object

SENSORS = [{'sensor': 'type.unity.com/unity.solo.RGBCamera', 'annotations': [<class 'pysolotools.core.models.solo.BoundingBox2DAnnotation'>, <class 'pysolotools.core.models.solo.BoundingBox3DAnnotation'>, <class 'pysolotools.core.models.solo.InstanceSegmentationAnnotation'>, <class 'pysolotools.core.models.solo.SemanticSegmentationAnnotation'>]}]

parse_frame(f_path: str) → Frame

Parses a json file to a pysolo Frame model.

Parameters: f_path (str) – Path to a step in a sequence for a frame.
Return type: Frame