tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-15 04:13:57 +08:00
Code Activity

Alignment of fast-exposed to well-exposed

Browse Source
This commit is contained in:
dotag1 committed 2024-08-15 16:32:19 -04:00
1 parent 8484fb9417
commit b3e2a391fd
10 files changed
+432 -92
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

lantern_scripts/create_sfm_folder.py
+7 -7
View File
@@ -4,21 +4,21 @@ import json
import yaml
import argparse
def copy_images(src_folder, dest_folder):
def copy_images(src_folder, dest_folder, folder):
os.makedirs(dest_folder, exist_ok=True)
files = os.listdir(src_folder)
for filename in files:
if filename.endswith(".png"):
src_file = os.path.join(src_folder, filename)
dest_file = os.path.join(dest_folder, filename)
dest_file = os.path.join(dest_folder, folder + "_" + filename)
shutil.copy(src_file, dest_file)
def create_camera_models_overrides_json(dest_folder):
camera_models_data = {
"all": {
"width": 3840,
"projection_type": "equirectangular",
"width": 3840,
"height": 1920
}
}
@@ -43,7 +43,7 @@ def create_config_yml(dest_folder):
if __name__ == '__main__':
argparser = argparse.ArgumentParser()
argparser.add_argument("--input_dir", type=str, default="/mnt/data/scene/")
argparser.add_argument("--input_dir", type=str, default="/mnt/data/scene")
args = argparser.parse_args()
src_folder = args.input_dir + "data/"
@@ -51,9 +51,9 @@ if __name__ == '__main__':
dest_image_folder = dest_folder + "images"
dest_mask_folder = dest_folder + "masks"
for folder in ("left_e1/", "left_sfm/", "right_sfm/"):
copy_images(src_folder + folder, dest_image_folder)
copy_images(src_folder + folder + "mask/", dest_mask_folder)
for folder in ("left_e1", "left_sfm", "right_sfm"):
copy_images(src_folder + folder + "/", dest_image_folder, folder)
copy_images(src_folder + folder + "/mask/", dest_mask_folder, folder)
create_camera_models_overrides_json(dest_folder)
create_config_yml(dest_folder)