datumaro.plugins.data_formats.kitti.format

Functions

make_kitti_categories([label_map])
parse_label_map(path)
write_label_map(path, label_map)

Classes

KittiPath()
KittiTask(value[, names, module, qualname, ...])

class datumaro.plugins.data_formats.kitti.format.KittiTask(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: Enum

segmentation = 1

detection = 2

class datumaro.plugins.data_formats.kitti.format.KittiPath

Bases: object

IMAGES_DIR = 'image_2'

INSTANCES_DIR = 'instance'

LABELS_DIR = 'label_2'

SEMANTIC_RGB_DIR = 'semantic_rgb'

SEMANTIC_DIR = 'semantic'

IMAGE_EXT = '.png'

MASK_EXT = '.png'

LABELMAP_FILE = 'label_colors.txt'

DEFAULT_TRUNCATED = 0.0

DEFAULT_OCCLUDED = 0

DEFAULT_SCORE = 1.0

datumaro.plugins.data_formats.kitti.format.make_kitti_categories(label_map=None)

datumaro.plugins.data_formats.kitti.format.parse_label_map(path)

datumaro.plugins.data_formats.kitti.format.write_label_map(path, label_map)