Skip to content

Commit cff9257

Browse files
authored
Merge pull request #1375 from jrobble:fix_caffe_swaprb
Set swapRB to false in GOTURN tracker
2 parents 8d0ed6e + 2fd0d74 commit cff9257

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

modules/tracking/src/gtrTracker.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -153,11 +153,11 @@ bool TrackerGOTURNImpl::updateImpl(const Mat& image, Rect2d& boundingBox)
153153
searchPatch = searchPatch - 128;
154154

155155
//Convert to Float type
156-
Mat targetBlob = dnn::blobFromImage(targetPatch);
157-
Mat searchBlob = dnn::blobFromImage(searchPatch);
156+
Mat targetBlob = dnn::blobFromImage(targetPatch, 1.0f, Size(), Scalar(), false);
157+
Mat searchBlob = dnn::blobFromImage(searchPatch, 1.0f, Size(), Scalar(), false);
158158

159-
net.setInput(targetBlob, ".data1");
160-
net.setInput(searchBlob, ".data2");
159+
net.setInput(targetBlob, "data1");
160+
net.setInput(searchBlob, "data2");
161161

162162
Mat resMat = net.forward("scale").reshape(1, 1);
163163

modules/tracking/test/test_trackers.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -532,7 +532,7 @@ TEST_P(DistanceAndOverlap, DISABLED_Scaled_Data_TLD)
532532

533533
TEST_P(DistanceAndOverlap, DISABLED_GOTURN)
534534
{
535-
TrackerTest test(TrackerGOTURN::create(), dataset, 0, 100, NoTransform);
535+
TrackerTest test(TrackerGOTURN::create(), dataset, 18, .5f, NoTransform);
536536
test.run();
537537
}
538538

0 commit comments

Comments
 (0)