5 using namespace suanzi;
9 double distance_cosine(const Eigen::VectorXd& u, const Eigen::VectorXd& v)
11 if (u.dot(u) * v.dot(v) == 0){
12 throw overflow_error("zero vector for cosine distance");
14 return (1 - u.dot(v) / std::sqrt(u.dot(u) * v.dot(v)));
17 double distance_euclidean(const Eigen::VectorXd& u, const Eigen::VectorXd& v)
19 Eigen::VectorXd d = u - v;
20 return std::sqrt(d.dot(d));
24 Rect getRectInDetection(const Detection& d)
27 r.x = d.center_x - d.width / 2;
28 r.y = d.center_y - d.height / 2;
34 double calc_iou_ratio(const Rect& r1, const Rect& r2)
37 Rect r_inner = r1 & r2;
38 Rect r_union = r1 | r2;
39 if (r_inner.area() > 0){
40 iou = 1.0 * r_inner.area() / r_union.area();
43 rr.x = min(r1.x + r1.width, r2.x + r2.width);
44 rr.y = min(r1.y + r1.height, r2.y + r2.height);
45 rr.width = max(r1.x, r2.x) - rr.x;
46 rr.height = max(r1.y, r2.y) - rr.y;
47 iou = -1.0 * rr.area() / r_union.area();