diff --git a/.gitignore b/.gitignore index 28afce358a..f7e1a29b51 100644 --- a/.gitignore +++ b/.gitignore @@ -190,4 +190,7 @@ camera_paths/ # pixi environments .pixi -/third_party \ No newline at end of file +/third_party + +# JetBrains IDEs +.idea diff --git a/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py b/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py index 55f657a3a4..ac3480a29a 100644 --- a/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py +++ b/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py @@ -29,9 +29,9 @@ class ColmapConverterToNerfstudioDataset(BaseConverterToNerfstudioDataset): """Base class to process images or video into a nerfstudio dataset using colmap""" - camera_type: Literal["perspective", "fisheye", "equirectangular", "pinhole", "simple_pinhole"] = "perspective" + camera_type: Literal["perspective", "fisheye", "equirectangular", "pinhole", "simple_pinhole"] = "simple_pinhole" """Camera model to use.""" - matching_method: Literal["exhaustive", "sequential", "vocab_tree"] = "vocab_tree" + matching_method: Literal["exhaustive", "sequential", "vocab_tree"] = "exhaustive" """Feature matching method to use. Vocab tree is recommended for a balance of speed and accuracy. Exhaustive is slower but more accurate. Sequential is faster but should only be used for videos.""" diff --git a/nerfstudio/process_data/colmap_utils.py b/nerfstudio/process_data/colmap_utils.py index 1d9405c81a..415d050cfa 100644 --- a/nerfstudio/process_data/colmap_utils.py +++ b/nerfstudio/process_data/colmap_utils.py @@ -126,7 +126,6 @@ def run_colmap( f"--image_path {image_dir}", "--ImageReader.single_camera 1", f"--ImageReader.camera_model {camera_model.value}", - f"--SiftExtraction.use_gpu {int(gpu)}", ] if camera_mask_path is not None: feature_extractor_cmd.append(f"--ImageReader.camera_mask_path {camera_mask_path}") @@ -140,7 +139,6 @@ def run_colmap( feature_matcher_cmd = [ f"{colmap_cmd} {matching_method}_matcher", f"--database_path {colmap_dir / 'database.db'}", - f"--SiftMatching.use_gpu {int(gpu)}", ] if matching_method == "vocab_tree": vocab_tree_filename = get_vocab_tree()