mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-08 01:53:19 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #25264 from lpylpy0514:4.x

Browse Source 
fix a bug about vittrack post-process
This commit is contained in:
Alexander Smorkalov 2024-03-28 09:53:29 +03:00 committed by GitHub
commit a132fd401e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/video/src/tracking/tracker_vit.cpp
View File

@ -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;