os.makedirs(DST_DETECTED_SKELETONS_FOLDER, exist_ok=True)
os.makedirs(DST_VIZ_IMGS_FOLDER, exist_ok=True)
# -- Read images and process
num_total_images = images_loader.num_images
for ith_img in range(num_total_images):
# -- Read image
img, str_action_label, img_info = images_loader.read_image()
# -- Detect
humans = skeleton_detector.detect(img)
# -- Draw
img_disp = img.copy()
skeleton_detector.draw(img_disp, humans)
# img_displayer.display(img_disp, wait_key_ms=1)
# -- Get skeleton data and save to file
skeletons, scale_h = skeleton_detector.humans_to_skels_list(humans)
dict_id2skeleton = multiperson_tracker.track(
skeletons) # dict: (int human id) -> (np.array() skeleton)
skels_to_save = [
img_info + skeleton.tolist()
for skeleton in dict_id2skeleton.values()
]
# -- Save result
# Save skeleton data for training
filename = SKELETON_FILENAME_FORMAT.format(ith_img)
os.makedirs(DST_DETECTED_SKELETONS_FOLDER, exist_ok=True)
os.makedirs(DST_VIZ_IMGS_FOLDER, exist_ok=True)
# -- Read images and process
num_total_images = images_loader.num_images
for ith_img in range(num_total_images):
# -- Read image
img, str_action_label, img_info = images_loader.read_image()
# -- Detect
humans = skeleton_detector.detect(img)
# -- Draw
img_disp = img.copy()
skeleton_detector.draw(img_disp, humans) # 在圖片上畫上骨骼
# 在colab要關掉
img_displayer.display(img_disp, wait_key_ms=1)
# -- Get skeleton data and save to file
skeletons, scale_h = skeleton_detector.humans_to_skels_list(humans)
dict_id2skeleton = multiperson_tracker.track(
skeletons) # dict: (int human id) -> (np.array() skeleton)
# {huamn id: human's skeleton}
skels_to_save = [
img_info + skeleton.tolist()
for skeleton in dict_id2skeleton.values()
]
# Save skeleton data for training