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)
