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)
