Merge pull request #25264 from lpylpy0514:4.x

fix a bug about vittrack post-process
pull/25291/head
Alexander Smorkalov 8 months ago committed by GitHub
commit a132fd401e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 4
      modules/video/src/tracking/tracker_vit.cpp

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

Loading…
Cancel
Save