From bcea394bd8fd45df04a859f2125385c89b2b75fa Mon Sep 17 00:00:00 2001
From: Realcat <alpharealcat@gmail.com>
Date: Fri, 4 Oct 2024 14:26:20 +0000
Subject: [PATCH] fix:
 https://github.com/Vincentqyw/image-matching-webui/issues/78

---
 ui/viz.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/ui/viz.py b/ui/viz.py
index e7fb8fb..d1d53c8 100644
--- a/ui/viz.py
+++ b/ui/viz.py
@@ -10,6 +10,10 @@
 
 from hloc.utils.viz import add_text, plot_keypoints
 
+np.random.seed(1995)
+color_map = np.arange(100)
+np.random.shuffle(color_map)
+
 
 def plot_images(
     imgs: List[np.ndarray],
@@ -232,11 +236,6 @@ def error_colormap(
     )
 
 
-np.random.seed(1995)
-color_map = np.arange(100)
-np.random.shuffle(color_map)
-
-
 def fig2im(fig: matplotlib.figure.Figure) -> np.ndarray:
     """
     Convert a matplotlib figure to a numpy array with RGB values.
@@ -284,9 +283,8 @@ def draw_matches_core(
         The figure as a numpy array with shape (height, width, 3) and dtype uint8
         containing the RGB values of the figure.
     """
-    thr = 5e-4
     thr = 0.5
-    color = error_colormap(conf, thr, alpha=0.1)
+    color = error_colormap(1 - conf, thr, alpha=0.1)
     text = [
         # "image name",
         f"#Matches: {len(mkpts0)}",