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

fix a bug about vittrack post-process

Browse Source
This commit is contained in:
lpylpy0514 2024-03-25 15:40:34 +08:00
parent 4c86b287fd
commit 84a1248b0a
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); preprocess(crop, blob, templateSize);
net.setInput(blob, "template"); net.setInput(blob, "template");
Size size(16, 16); Size size(16, 16);
hanningWindow = hann2d(size, false); hanningWindow = hann2d(size, true);
rect_last = boundingBox_; 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 size_map = outs[1].reshape(0, {2, 16, 16});
Mat offset_map = outs[2].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; double maxVal;
Point maxLoc; Point maxLoc;