pose_estimation/extract_pose.py

import os
import shutil
import sys
from glob import glob

gpu_id = sys.argv[1]
custom_folder = sys.argv[2]
temp_folder = sys.argv[3]
output_folder = sys.argv[4]

name_list = [x for x in sorted(glob("%s/*.png"%(custom_folder))) if 'mask' not in x]
os.system('CUDA_VISIBLE_DEVICES=%s python DataProcess/Gen_HeadMask.py --img_dir ./%s/'%(gpu_id,custom_folder))
os.system('CUDA_VISIBLE_DEVICES=%s python DataProcess/Gen_Landmark.py --img_dir ./%s/'%(gpu_id,custom_folder))
for name_all in name_list:
    name = os.path.basename(name_all)[:-4]
    os.system('python align_roll.py %s %s %s'%(name, custom_folder,temp_folder))

os.system('CUDA_VISIBLE_DEVICES=%s python process_test_images.py --input_dir ./%s/ --gpu=%s'%(gpu_id,temp_folder,gpu_id))

os.system('python check_pose.py %s %s '%(temp_folder,output_folder))

##example
#python extract_pose.py 0 custom_imgs_folder temp_folder output_folder