Use --image_list / --feature_path arguments.

cvg · Feb 21, 2024 · e43ce1b · e43ce1b
1 parent e3e953f
commit e43ce1b
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/hloc/extract_features.py b/hloc/extract_features.py
@@ -307,4 +307,11 @@ def main(
     parser.add_argument("--image_list", type=Path)
     parser.add_argument("--feature_path", type=Path)
     args = parser.parse_args()
-    main(confs[args.conf], args.image_dir, args.export_dir, args.as_half)
+    main(
+        confs[args.conf],
+        args.image_dir,
+        args.export_dir,
+        args.as_half,
+        args.image_list,
+        args.feature_path,
+    )