Fix: #54 (#62)

Vincentqyw · Aug 21, 2024 · 07c30b9 · 07c30b9
1 parent e4a103b
commit 07c30b9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ui/utils.py b/ui/utils.py
@@ -910,8 +910,9 @@ def run_matching(
     t1 = time.time()
 
     if model["dense"]:
-        if not match_conf["preprocessing"]["force_resize"]:
+        if not match_conf["preprocessing"].get("force_resize", False):
             match_conf["preprocessing"]["force_resize"] = force_resize
+        else:
             logger.info("preprocessing is already resized")
         if force_resize:
             match_conf["preprocessing"]["height"] = image_height
@@ -941,7 +942,7 @@ def run_matching(
         else:
             extractor = get_feature_model(extract_conf)
 
-        if not extract_conf["preprocessing"]["force_resize"]:
+        if not extract_conf["preprocessing"].get("force_resize", False):
             extract_conf["preprocessing"]["force_resize"] = force_resize
         else:
             logger.info("preprocessing is already resized")