fantasyfish/dreamgaussian

1import os2import glob3import argparse4 5parser = argparse.ArgumentParser()6parser.add_argument('--dir', default='logs', type=str, help='Directory where obj files are stored')7parser.add_argument('--out', default='videos', type=str, help='Directory where videos will be saved')8args = parser.parse_args()9 10out = args.out11os.makedirs(out, exist_ok=True)12 13files = glob.glob(f'{args.dir}/*.obj')14for f in files:15    name = os.path.basename(f)16    # first stage model, ignore17    if name.endswith('_mesh.obj'): 18        continue19    print(f'[INFO] process {name}')20    os.system(f"python -m kiui.render {f} --save_video {os.path.join(out, name.replace('.obj', '.mp4'))} ")