zokyo.augmentation.utils module

zokyo.augmentation.utils.apply_augmentation(image, mask, label, function)

Function to apply augmentation operation to a certain labels only

zokyo.augmentation.utils.change_pascal_annotation(annotation, image_dir, filename)

Function add path infos to pascal annotation

zokyo.augmentation.utils.generate_mask_for_annotation_for_xml(annotation, num_classes, label_id)

Method to generate class-wise binary mask from the bounding boxes of each class (including BG)

zokyo.augmentation.utils.get_annotation_dictionary(annotation)

Method to parse XML annotation and return a dict with class names as keys and their corresponding bounding boxes as values.