Skip to content

Commit 840980d

Browse files
Skelpmerll002
andauthored
Correct GPU feature extraction option (#1)
* Correct GPU feature extraction option * Update flag --------- Co-authored-by: Leo <76182388+merll002@users.noreply.github.com>
1 parent 50e0e3c commit 840980d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

nerfstudio/process_data/colmap_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def run_colmap(
126126
f"--image_path {image_dir}",
127127
"--ImageReader.single_camera 1",
128128
f"--ImageReader.camera_model {camera_model.value}",
129-
f"--SiftExtraction.use_gpu {int(gpu)}",
129+
f"--FeatureExtraction.use_gpu {int(gpu)}",
130130
]
131131
if camera_mask_path is not None:
132132
feature_extractor_cmd.append(f"--ImageReader.camera_mask_path {camera_mask_path}")
@@ -140,7 +140,7 @@ def run_colmap(
140140
feature_matcher_cmd = [
141141
f"{colmap_cmd} {matching_method}_matcher",
142142
f"--database_path {colmap_dir / 'database.db'}",
143-
f"--SiftMatching.use_gpu {int(gpu)}",
143+
f"--FeatureMatching.use_gpu {int(gpu)}",
144144
]
145145
if matching_method == "vocab_tree":
146146
vocab_tree_filename = get_vocab_tree()

0 commit comments

Comments
 (0)