Skip to content

Bounding box fixes for GOTURN tracker #2620

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
alalek merged 18 commits into from
Sep 5, 2020
Merged

Bounding box fixes for GOTURN tracker #2620

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
735511b
Merge pull request #2 from opencv/3.4
ieliz Aug 14, 2020
63c3cdf
Trying to change branch from master to 3.4
ieliz Aug 14, 2020
5ab85c1
Removing unnecessary if statement
ieliz Aug 14, 2020
2dac64a
Frame existence check is added
ieliz Aug 17, 2020
0d5b40a
Changes in GOTURN tracker for reading path to caffemodel and prototxt…
ieliz Aug 19, 2020
229653a
Replacing get_data_path with findDataFile and moving paths in parameters
ieliz Aug 19, 2020
81b3c20
tracking: fix GOTURN model loading
alalek Aug 26, 2020
8146256
Fixing GOTURN memory test: paths to GOTURN files are in cv::FileNode now
ieliz Sep 3, 2020
115a52e
Fix
ieliz Sep 3, 2020
3707382
Fix
ieliz Sep 3, 2020
840992d
Fixing comments
ieliz Sep 3, 2020
b066fd8
Deleting trailing space
ieliz Sep 4, 2020
ca9e474
Trying to solve the problem with test
ieliz Sep 4, 2020
7c1148c
Removing ground truth reading part in GOTURN test
ieliz Sep 4, 2020
743b5e0
Removing trailing spaces
ieliz Sep 4, 2020
e54e573
Trying to fix win64/win32 issue
ieliz Sep 4, 2020
6303fb9
Fixing problem with win32/win64
ieliz Sep 4, 2020
447cf39
Trying to solve issue with win32/win64
ieliz Sep 4, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion modules/tracking/src/gtrTracker.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,9 @@ class TrackerGOTURNModel : public TrackerModel{
public:
TrackerGOTURNModel(TrackerGOTURN::Params){}
Rect2d getBoundingBox(){ return boundingBox_; }
void setBoudingBox(Rect2d boundingBox){ boundingBox_ = boundingBox; }
void setBoudingBox(Rect2d boundingBox) {
boundingBox_ = boundingBox & Rect2d(Point(0, 0), image_.size());;
}
Mat getImage(){ return image_; }
void setImage(const Mat& image){ image.copyTo(image_); }
protected:
Expand Down Expand Up @@ -137,6 +139,11 @@ bool TrackerGOTURNImpl::updateImpl(const Mat& image, Rect2d& boundingBox)
targetPatchRect.x = (float)(prevCenter.x - prevBB.width*padTargetPatch / 2.0 + targetPatchRect.width);
targetPatchRect.y = (float)(prevCenter.y - prevBB.height*padTargetPatch / 2.0 + targetPatchRect.height);

targetPatchRect.width = std::min(targetPatchRect.width, (float)prevFrame.cols);
targetPatchRect.height = std::min(targetPatchRect.height, (float)prevFrame.rows);
targetPatchRect.x = std::max(-prevFrame.cols * 0.5f, std::min(targetPatchRect.x, prevFrame.cols * 1.5f));
targetPatchRect.y = std::max(-prevFrame.rows * 0.5f, std::min(targetPatchRect.y, prevFrame.rows * 1.5f));

copyMakeBorder(prevFrame, prevFramePadded, (int)targetPatchRect.height, (int)targetPatchRect.height, (int)targetPatchRect.width, (int)targetPatchRect.width, BORDER_REPLICATE);
targetPatch = prevFramePadded(targetPatchRect).clone();

Expand Down
16 changes: 16 additions & 0 deletions modules/tracking/test/test_trackers.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -567,6 +567,22 @@ TEST_P(DistanceAndOverlap, Scaled_Data_CSRT)
test.run();
}

TEST(GOTURN, memory_usage)
{
cv::Rect2d roi(145, 70, 85, 85);
cv::Mat frame;
cv::Ptr<cv::Tracker> tracker = cv::TrackerGOTURN::create();
string folder = cvtest::TS::ptr()->get_data_path() + "tracking/david/data/david.webm";
cv::VideoCapture video( folder );
video >> frame;
tracker->init(frame, roi);
for (;; ) {
video >> frame;
if (frame.empty())
break;
tracker->update(frame, roi);
}
}

INSTANTIATE_TEST_CASE_P( Tracking, DistanceAndOverlap, TESTSET_NAMES);

Expand Down