def combine_dataset_from_config_file(config_path: str, dest_folder_img: str, dest_json_file: str): # combine dataset combined_dataset = COCO_Dataset.combine_from_config( config_path=config_path, img_sort_attr_name='file_name', show_pbar=True) combined_dataset.move_images(dst_img_dir=dest_folder_img, preserve_filenames=False, update_img_paths=True, overwrite=True, show_pbar=True) combined_dataset.save_to_path(save_path=dest_json_file, overwrite=True)
# logger.cyan(info1.contributor) # logger.purple(f'info:\n{info}') # logger.purple(f'info0:\n{info0}') # logger.purple(f'info.to_dict():\n{info.to_dict()}') info2 = COCO_Info.from_dict({ 'description': 'This is a test', 'url': 'https://test/url.com', 'version': '1.0', 'year': '2020', 'contributor': 'Clayton', 'date_created': '2020/03/10' }) logger.purple(f'info2:\n{info2}') info2.save_to_path('info.json', overwrite=True) from annotation_utils.coco.structs import COCO_Dataset dataset = COCO_Dataset.combine_from_config( config_path= '/home/clayton/workspace/prj/data_keep/data/toyota/dataset/config/yaml/box_hsr_kpt_sim.yaml', img_sort_attr_name='file_name', show_pbar=True) # dataset.display_preview(kpt_idx_offset=-1) coco_ann = dataset.annotations[123] logger.purple(coco_ann) from common_utils.common_types.keypoint import Keypoint3D_List coco_ann.keypoints_3d = Keypoint3D_List() logger.purple(f'coco_ann.to_dict():\n{coco_ann.to_dict()}')
from annotation_utils.coco.structs import COCO_Dataset # dataset = COCO_Dataset.load_from_path( # json_path='/home/clayton/workspace/prj/data_keep/data/toyota/dataset/sim/20200228/28_02_2020_11_18_30_coco-data/HSR-coco.json', # img_dir='/home/clayton/workspace/prj/data_keep/data/toyota/dataset/sim/20200228/28_02_2020_11_18_30_coco-data' # ) dataset = COCO_Dataset.combine_from_config( '/home/clayton/workspace/prj/data_keep/data/toyota/dataset/config/json/box_hsr_kpt_real.json' ) dataset.images.sort(attr_name='file_name') dataset.save_video(save_path='dataset.mp4', show_annotations=True, fps=5, show_preview=True, kpt_idx_offset=-1, overwrite=True, show_bbox=False)