def get_pose_data_list(data_path, metas_filename, min_count, min_score):
"""
data_path : image and anno folder name
"""
print("[x] Get pose data from {}".format(data_path))
data = PoseInfo(data_path, metas_filename, min_count, min_score)
rgbs_file_list, depths_file_list, anno2ds_list = data.get_2d_data_list()
if len(rgbs_file_list) != len(anno2ds_list) or len(depths_file_list) != len(anno2ds_list):
raise Exception("number of images, cameras and annotations do not match")
else:
print("{} has {} groups of images".format(data_path, len(rgbs_file_list)))
return rgbs_file_list, depths_file_list, anno2ds_list
def get_pose_data_list(im_path, ann_path):
"""
train_im_path : image folder name
train_ann_path : coco json file name
"""
print("[x] Get pose data from {}".format(im_path))
data = PoseInfo(im_path, ann_path, False)
imgs_file_list = data.get_image_list()
objs_info_list = data.get_joint_list()
mask_list = data.get_mask()
targets = list(zip(objs_info_list, mask_list))
if len(imgs_file_list) != len(objs_info_list):
raise Exception("number of images and annotations do not match")
else:
print("{} has {} images".format(im_path, len(imgs_file_list)))
return imgs_file_list, objs_info_list, mask_list, targets
def get_files(data_path):
sum_rgbs_list, sum_depths_list, sum_cams_list = [], [], []
root_list = tl.files.load_folder_list(path=data_path)
for root in root_list:
folder_list = tl.files.load_folder_list(path=root)
for folder in folder_list:
if 'KINECTNODE' in folder:
print("[x] Get pose data from {}".format(folder))
_cams_list, _rgbs_list, _depths_list, _ = PoseInfo(
folder, 'meta.mat', 9, 0.25).get_val_data_list()
sum_cams_list.extend(_cams_list)
sum_rgbs_list.extend(_rgbs_list)
sum_depths_list.extend(_depths_list)
print("Total number of own images found:", len(sum_rgbs_list))
return sum_rgbs_list, sum_depths_list, sum_cams_list