From 84474794f0391a958122891dc79c65c922e133f3 Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Mon, 13 Nov 2023 13:03:34 +0800
Subject: [PATCH] Fix: Remove secondary peaks in position results

---
 tasks/map/minimap/minimap.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/tasks/map/minimap/minimap.py b/tasks/map/minimap/minimap.py
index 4b7d14edf..86fd52f2f 100644
--- a/tasks/map/minimap/minimap.py
+++ b/tasks/map/minimap/minimap.py
@@ -133,17 +133,21 @@ class Minimap(MapResource):
         result_mask = image_center_crop(result_mask, size=image_size(result))
         result[result_mask] = 0
         _, sim, _, loca = cv2.minMaxLoc(result)
+        # from PIL import Image
         # if round(scale, 3) == self.POSITION_SEARCH_SCALE * 1.0:
         #     result[result <= 0] = 0
         #     Image.fromarray((result * 255).astype(np.uint8)).save('match_result.png')
 
         # Gaussian filter to get local maximum
-        local_maximum = cv2.subtract(result, cv2.GaussianBlur(result, (5, 5), 0))
+        local_maximum = subtract_blur(result, radius=5)
+        # Multiply to remove secondary peaks
+        cv2.multiply(local_maximum, result, dst=local_maximum)
+        cv2.multiply(local_maximum, 10, dst=local_maximum)
         _, local_sim, _, local_loca = cv2.minMaxLoc(local_maximum)
         # if round(scale, 5) == self.POSITION_SEARCH_SCALE * 1.0:
         #     local_maximum[local_maximum < 0] = 0
-        #     local_maximum[local_maximum > 0.1] = 0.1
-        #     Image.fromarray((local_maximum * 255 * 10).astype(np.uint8)).save('local_maximum.png')
+        #     local_maximum[local_maximum > 1] = 1
+        #     Image.fromarray((local_maximum * 255).astype(np.uint8)).save('local_maximum.png')
 
         # Calculate the precise location using CUBIC
         # precise = crop(result, area=area_offset((-4, -4, 4, 4), offset=local_loca))