Spaces:
Running
on
A10G
Running
on
A10G
File size: 666 Bytes
12b7f59 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import os
import glob
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--dir', default='logs', type=str, help='Directory where obj files are stored')
parser.add_argument('--out', default='videos', type=str, help='Directory where videos will be saved')
args = parser.parse_args()
out = args.out
os.makedirs(out, exist_ok=True)
files = glob.glob(f'{args.dir}/*.obj')
for f in files:
name = os.path.basename(f)
# first stage model, ignore
if name.endswith('_mesh.obj'):
continue
print(f'[INFO] process {name}')
os.system(f"python -m kiui.render {f} --save_video {os.path.join(out, name.replace('.obj', '.mp4'))} ") |