from annotation_utils.labelme.structs import LabelmeAnnotationHandler from annotation_utils.coco.structs import COCO_Dataset, COCO_Category_Handler # Path Variables img_dir = '' json_dir = '' categories_conf_path = '' # Load Labelme Annotation Handler labelme_handler = LabelmeAnnotationHandler.load_from_dir(load_dir=json_dir) # Load COCO Dataset from Labelme Annotation Handler coco_dataset = COCO_Dataset.from_labelme( labelme_handler=labelme_handler, categories=COCO_Category_Handler.load_from_path(categories_conf_path), img_dir=img_dir ) # Save COCO Dataset to a file coco_dataset.save_to_path(save_path='', overwrite=True) # Preview COCO Dataset coco_dataset.display_preview(show_details=True) coco_dataset.save_visualization(show_details=True) # coco_dataset.save_video(show_details=True)
categories.append( # Simple Keypoint Example COCO_Category.from_label_skeleton( id=len(categories), supercategory='pet', name='cat', label_skeleton=[ ['left_eye', 'right_eye'], ['mouth_left', 'mouth_center'], ['mouth_center', 'mouth_right'] ] ) ) for name in ['duck', 'sparrow', 'pigion']: categories.append( # Simple Non-Keypoint Example COCO_Category( id=len(categories), supercategory='bird', name=name ) ) categories.save_to_path('categories_example.json') else: # Or load from an existing categories json categories = COCO_Category_Handler.load_from_path('categories_example.json') # Convert To COCO coco_dataset = COCO_Dataset.from_labelme( labelme_handler=labelme_handler, categories=categories, img_dir=img_dir ) coco_dataset.save_to_path(save_path='converted_coco.json', overwrite=True) coco_dataset.display_preview(show_details=True) # Optional: Preview your resulting dataset.
from annotation_utils.labelme.structs import LabelmeAnnotationHandler from annotation_utils.coco.structs import COCO_Dataset, COCO_Category_Handler img_dir = '/home/clayton/workspace/prj/data_keep/data/toyota/dataset/real/phone_videos/new/sampled_data/VID_20200217_161043/img' json_dir = '/home/clayton/workspace/prj/data_keep/data/toyota/dataset/real/phone_videos/new/sampled_data/VID_20200217_161043/json' labelme_handler = LabelmeAnnotationHandler.load_from_dir(load_dir=json_dir) coco_dataset = COCO_Dataset.from_labelme( labelme_handler=labelme_handler, categories=COCO_Category_Handler.load_from_path( '/home/clayton/workspace/prj/data_keep/data/toyota/dataset/config/categories/hsr_categories.json' ), img_dir=img_dir) coco_dataset.save_to_path(save_path='output.json', overwrite=True) test_coco_dataset = COCO_Dataset.load_from_path(json_path='output.json') test_coco_dataset.display_preview(kpt_idx_offset=-1)