|
|
|
|
|
import numpy as np |
|
|
|
import os |
|
import argparse |
|
|
|
parser = argparse.ArgumentParser() |
|
parser.add_argument('-d', '--openpose_dir', type=str, required=True) |
|
parser.add_argument('-i', '--input_root', type=str, required=True) |
|
parser.add_argument('-o', '--out_path', type=str, required=True) |
|
args = parser.parse_args() |
|
|
|
op_dir = args.openpose_dir |
|
input_path = args.input_root |
|
out_json_path = args.out_path |
|
|
|
os.makedirs(out_json_path, exist_ok=True) |
|
|
|
cmd = "cd {0}; ./build/examples/openpose/openpose.bin --image_dir {1} --write_json {2} --render_pose 2 --face --face_render 2 --hand --hand_render 2".format(op_dir, input_path, out_json_path) |
|
print(cmd) |
|
os.system(cmd) |
|
|
