From 84a1248b0a0016f75caa9cf056caf098394e3b9e Mon Sep 17 00:00:00 2001
From: lpylpy0514 <1872918507@qq.com>
Date: Mon, 25 Mar 2024 15:40:34 +0800
Subject: [PATCH] fix a bug about vittrack post-process

---
 modules/video/src/tracking/tracker_vit.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/video/src/tracking/tracker_vit.cpp b/modules/video/src/tracking/tracker_vit.cpp
index b807948033..bef42dbb4d 100644
--- a/modules/video/src/tracking/tracker_vit.cpp
+++ b/modules/video/src/tracking/tracker_vit.cpp
@@ -163,7 +163,7 @@ void TrackerVitImpl::init(InputArray image_, const Rect &boundingBox_)
     preprocess(crop, blob, templateSize);
     net.setInput(blob, "template");
     Size size(16, 16);
-    hanningWindow = hann2d(size, false);
+    hanningWindow = hann2d(size, true);
     rect_last = boundingBox_;
 }
 
@@ -184,7 +184,7 @@ bool TrackerVitImpl::update(InputArray image_, Rect &boundingBoxRes)
     Mat size_map = outs[1].reshape(0, {2, 16, 16});
     Mat offset_map = outs[2].reshape(0, {2, 16, 16});
 
-    multiply(conf_map, (1.0 - hanningWindow), conf_map);
+    multiply(conf_map, hanningWindow, conf_map);
 
     double maxVal;
     Point maxLoc;