no GPU for feature extraction too

ErlerPhilipp · Oct 1, 2024 · 96447d0 · 96447d0
1 parent a41db0a
commit 96447d0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/images_to_mesh/processing_steps/sfm/reconstruct.py b/images_to_mesh/processing_steps/sfm/reconstruct.py
@@ -63,7 +63,8 @@ def reconstruct_with_colmap_sparse(image_list: List[str]) -> List[str]:
         'colmap', 'feature_extractor',
         '--database_path', database_path,
         '--image_path', image_path,
-        '--SiftExtraction.num_threads', num_processes
+        '--SiftExtraction.num_threads', num_processes,
+        '--use_gpu', 0
     ]
     try:
         for line in execute_subprocess(command=extract_command, log_path=log_path, error_path=error_path):