projects
/
trackerpp.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
complete the metris functions
[trackerpp.git]
/
src
/
MultiTracker.cpp
diff --git
a/src/MultiTracker.cpp
b/src/MultiTracker.cpp
index
0b4dab5
..
c87395e
100644
(file)
--- a/
src/MultiTracker.cpp
+++ b/
src/MultiTracker.cpp
@@
-31,10
+31,23
@@
MultiTracker::~MultiTracker()
trackers.clear();
}
trackers.clear();
}
+static Rect getRectInDetection(const Detection& d)
+{
+ Rect r;
+ r.x = d.center_x - d.width / 2;
+ r.y = d.center_y - d.height / 2;
+ r.width = d.width;
+ r.height = d.height;
+ return r;
+}
+
static double calc_iou_ratio(const Detection& d1, const Detection& d2)
{
static double calc_iou_ratio(const Detection& d1, const Detection& d2)
{
- // TODO
- return 0.1;
+ Rect r1 = getRectInDetection (d1);
+ Rect r2 = getRectInDetection (d2);
+ Rect r_inner = r1 & r1;
+ Rect r_union = r1 | r2;
+ return 1.0 * r_inner.area() / r_union.area();
}
static std::vector<double> similarity(const PatchPtr p1, const PatchPtr p2)
}
static std::vector<double> similarity(const PatchPtr p1, const PatchPtr p2)
@@
-66,8
+79,7
@@
static std::vector<double> similarity(const PatchPtr p1, const PatchPtr p2)
double center_distance = sqrt(pow((d1.center_x - d2.center_x), 2) + pow((d1.center_y - d2.center_y), 2));
feature.push_back(center_distance / (d1.width + d1.height + d2.width + d2.height) * 4);
double center_distance = sqrt(pow((d1.center_x - d2.center_x), 2) + pow((d1.center_y - d2.center_y), 2));
feature.push_back(center_distance / (d1.width + d1.height + d2.width + d2.height) * 4);
- double iou_ratio = calc_iou_ratio(d1, d2);
- feature.push_back(iou_ratio);
+ feature.push_back(calc_iou_ratio(d1, d2));
return feature;
}
return feature;
}
@@
-141,12
+153,9
@@
void MultiTracker::update(unsigned int total, const Detection* detections, const
PatchPtr pp = createPatch(image, dd);
}
PatchPtr pp = createPatch(image, dd);
}
-// Get image crop from input image within given bounding box - Detecinon
static cv::Mat image_crop(const cv::Mat& image, const Detection& bb)
{
static cv::Mat image_crop(const cv::Mat& image, const Detection& bb)
{
- // RECT
- // TODO;
- return image.clone();
+ return image(getRectInDetection(bb));
}
PatchPtr MultiTracker::createPatch(const Mat& image, const Detection& detect)
}
PatchPtr MultiTracker::createPatch(const Mat& image, const Detection& detect)