mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 19:04:03 +08:00
55 lines
2.7 KiB
Python
55 lines
2.7 KiB
Python
import subprocess
|
|
import argparse
|
|
import os
|
|
import shutil
|
|
|
|
|
|
if __name__ == '__main__':
|
|
argparser = argparse.ArgumentParser()
|
|
argparser.add_argument("--input_dir", type=str, required=True, default="/mnt/data/scene/", help="Directory containing all the 4 videos")
|
|
argparser.add_argument("--shutter_speed", type=float, required=True, default=0.003125)
|
|
argparser.add_argument("--gt", action='store_true', default=False, help="Process GT images for metrics if flag is present")
|
|
args = argparser.parse_args()
|
|
|
|
subprocess.run(["python", "lantern_scripts/sync_and_trim.py", "--input_dir", args.input_dir], check=True)
|
|
subprocess.run(["python", "lantern_scripts/mp4_videos_to_png_frames.py", "--input_dir", args.input_dir + "data/"], check=True)
|
|
subprocess.run(["python", "lantern_scripts/mask_humans2.py", "--input_dir", args.input_dir + "data/", "--output_dir", args.input_dir + "data/masks"], check=True)
|
|
subprocess.run(["python", "lantern_scripts/create_data_folder.py", "--input_dir", args.input_dir + "data/", "--shutter_speed", str(args.shutter_speed)], check=True)
|
|
subprocess.run(["python", "lantern_scripts/create_sfm_folder.py", "--input_dir", args.input_dir], check=True)
|
|
|
|
# For GT
|
|
if args.gt:
|
|
subprocess.run(["python", "lantern_scripts/debevec_bracket_merge.py", "--data_dir", args.input_dir + "GT/", "--experiment_location", args.input_dir + "data/"], check=True)
|
|
subprocess.run(["python", "lantern_scripts/create_GT_folder.py", "--input_dir", args.input_dir + "GT/"], check=True)
|
|
|
|
gt_dir = os.path.join(args.input_dir, "GT")
|
|
gt_exr = os.path.join(gt_dir, "GT_exr")
|
|
gt_exr_renders = os.path.join(gt_dir, "GT_exr_renders")
|
|
|
|
print("Rendering GT images with Blender...")
|
|
subprocess.run([
|
|
"blender", "--background", "lantern_scripts/scene_new.blend",
|
|
"-P", "lantern_scripts/hdr_blender.py", "--",
|
|
os.path.join(gt_exr, ""), os.path.join(gt_exr_renders, "")
|
|
], check=True)
|
|
|
|
print("Converting EXR panos to LDR visualization...")
|
|
subprocess.run([
|
|
"python", "lantern_scripts/tonemap.py",
|
|
"--data_dir", gt_exr,
|
|
"--out_dir", gt_exr
|
|
], check=True)
|
|
subprocess.run([
|
|
"python", "lantern_scripts/tonemap.py",
|
|
"--data_dir", gt_exr_renders,
|
|
"--out_dir", gt_exr_renders
|
|
], check=True)
|
|
|
|
subprocess.run(["python", "lantern_scripts/gt_visualization.py",
|
|
"--gt_dir", gt_exr,
|
|
"--gt_render_dir", gt_exr_renders,
|
|
"--out_dir", args.input_dir
|
|
], check=True)
|
|
|
|
subprocess.run(["python", "lantern_scripts/data_visualization.py", "--input_dir", args.input_dir + "data/"], check=True)
|