src/Utils.cpp

   1 #include "Utils.h"
   2
   3 using namespace suanzi;
   4 using namespace cv;
   5
   6 double distance_cosine(const Eigen::VectorXd& u, const Eigen::VectorXd& v)
   7 {
   8     return (1 - u.dot(v) / std::sqrt(u.dot(u) * v.dot(v)));
   9 }
  10
  11 double distance_euclidean(const Eigen::VectorXd& u, const Eigen::VectorXd& v)
  12 {
  13     Eigen::VectorXd d = u - v;
  14     return std::sqrt(d.dot(d));
  15 }
  16
  17
  18 Rect getRectInDetection(const Detection& d)
  19 {
  20     Rect r;
  21     r.x = d.center_x - d.width / 2;
  22     r.y = d.center_y - d.height / 2;
  23     r.width = d.width;
  24     r.height = d.height;
  25     return r;
  26 }
  27
  28 double calc_iou_ratio(const Rect& r1, const Rect& r2)
  29 {
  30     double iou;
  31     Rect r_inner = r1 & r2;
  32     Rect r_union  = r1 | r2;
  33     if (r_inner.area() > 0){
  34         iou = 1.0 * r_inner.area() / r_union.area();
  35     }else{
  36         Rect rr;
  37         rr.x = min(r1.x + r1.width, r2.x + r2.width);
  38         rr.y = min(r1.y + r1.height, r2.y + r2.height);
  39         rr.width = max(r1.x, r2.x) - rr.x;
  40         rr.height = max(r1.y, r2.y) - rr.y;
  41         iou = -1.0 * rr.area() / r_union.area();
  42     }
  43     return iou;
  44
  45 }