Spaces:
Build error
Build error
import argparse | |
import os | |
from visualize import vis_utils | |
import shutil | |
from tqdm import tqdm | |
if __name__ == '__main__': | |
parser = argparse.ArgumentParser() | |
parser.add_argument("--input_path", type=str, required=True, help='stick figure mp4 file to be rendered.') | |
parser.add_argument("--cuda", type=bool, default=True, help='') | |
parser.add_argument("--device", type=int, default=0, help='') | |
params = parser.parse_args() | |
assert params.input_path.endswith('.mp4') | |
parsed_name = os.path.basename(params.input_path).replace('.mp4', '').replace('sample', '').replace('rep', '') | |
sample_i, rep_i = [int(e) for e in parsed_name.split('_')] | |
npy_path = os.path.join(os.path.dirname(params.input_path), 'results.npy') | |
out_npy_path = params.input_path.replace('.mp4', '_smpl_params.npy') | |
assert os.path.exists(npy_path) | |
results_dir = params.input_path.replace('.mp4', '_obj') | |
if os.path.exists(results_dir): | |
shutil.rmtree(results_dir) | |
os.makedirs(results_dir) | |
npy2obj = vis_utils.npy2obj(npy_path, sample_i, rep_i, | |
device=params.device, cuda=params.cuda) | |
print('Saving obj files to [{}]'.format(os.path.abspath(results_dir))) | |
for frame_i in tqdm(range(npy2obj.real_num_frames)): | |
npy2obj.save_obj(os.path.join(results_dir, 'frame{:03d}.obj'.format(frame_i)), frame_i) | |
print('Saving SMPL params to [{}]'.format(os.path.abspath(out_npy_path))) | |
npy2obj.save_npy(out_npy_path) | |