include/Utils.h

   1 #ifndef _UTILS_H_
   2 #define _UTILS_H_
   3
   4 #include <eigen3/Eigen/Dense>
   5 #include "Detector.h"
   6
   7
   8 // Computes the consine distance between u and v
   9 // https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.distance.cosine.html#scipy.spatial.distance.cosine
  10 double distance_cosine(const Eigen::VectorXd& u, const Eigen::VectorXd& v);
  11
  12 // Computes  the Euclidean distance between u and v
  13 // https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.distance.euclidean.html#scipy.spatial.distance.euclidean
  14 double distance_euclidean(const Eigen::VectorXd& u, const Eigen::VectorXd& v);
  15
  16
  17
  18 // Convert coordinates in Detection (center_x, center_y, w, h) to cv::Rect (x, y, w, h)
  19 cv::Rect getRectInDetection(const suanzi::Detection& d);
  20
  21
  22 // Calculate the IOU (Intersection Over Union) of two Detection
  23 double calc_iou_ratio(const cv::Rect& r1, const cv::Rect& r2);
  24
  25 #endif // _UTILS_H_