UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2024-11-25 21:28:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

scene alignment

Browse Source
This commit is contained in:
Chris Heinrich 2023-10-01 13:41:36 -07:00
parent 414b553ef1
commit 14ff61c768
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
convert.py
View File

@ -18,6 +18,7 @@ import shutil
parser = ArgumentParser("Colmap converter") parser = ArgumentParser("Colmap converter")
parser.add_argument("--no_gpu", action='store_true') parser.add_argument("--no_gpu", action='store_true')
parser.add_argument("--skip_matching", action='store_true') parser.add_argument("--skip_matching", action='store_true')
parser.add_argument("--skip_alignment", action='store_true')
parser.add_argument("--source_path", "-s", required=True, type=str) parser.add_argument("--source_path", "-s", required=True, type=str)
parser.add_argument("--camera", default="OPENCV", type=str) parser.add_argument("--camera", default="OPENCV", type=str)
parser.add_argument("--colmap_executable", default="", type=str) parser.add_argument("--colmap_executable", default="", type=str)
@ -65,6 +66,20 @@ if not args.skip_matching:
logging.error(f"Mapper failed with code {exit_code}. Exiting.") logging.error(f"Mapper failed with code {exit_code}. Exiting.")
exit(exit_code) exit(exit_code)
if not args.skip_alignment:
### Model alignment
# Aligns the scene so that +Z axis is up.
model_alignment_cmd = (colmap_command + " model_aligner \
--input_path " + args.source_path + "/distorted/sparse/0 \
--database_path " + args.source_path + "/distorted/database.db \
--alignment_max_error " + "0.1" "\
--alignment_type " + "plane" "\
--output_path " + args.source_path + "/distorted/sparse/0")
exit_code = os.system(model_alignment_cmd)
if exit_code != 0:
logging.error(f"Model aligner failed with code {exit_code}. Exiting.")
exit(exit_code)
### Image undistortion ### Image undistortion
## We need to undistort our images into ideal pinhole intrinsics. ## We need to undistort our images into ideal pinhole intrinsics.
img_undist_cmd = (colmap_command + " image_undistorter \ img_undist_cmd = (colmap_command + " image_undistorter \