include/hungarian.h

   1 #ifndef _HUNGARIAN_H_
   2 #define _HUNGARIAN_H_
   3 #include <eigen3/Eigen/Dense>
   4
   5 // C++ version of linear_sum_assignment() in python scipy
   6 // https://docs.scipy.org/doc/scipy-0.18.1/reference/generated/scipy.optimize.linear_sum_assignment.html
   7 // http://csclab.murraystate.edu/~bob.pilgrim/445/munkres.html
   8 //
   9
  10 // @param cost_matrix The cost matrix
  11 // @param row_ind, col_ind an array of row indices and one of corresponding column indices giving the optimal assignment.
  12 // @return the cost of the assignment
  13 int linear_sum_assignment(const Eigen::MatrixXi& cost_matrix, Eigen::VectorXi& row_ind, Eigen::VectorXi& col_ind);
  14
  15
  16 // Computes the consine distance between u and v
  17 // https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.distance.cosine.html#scipy.spatial.distance.cosine
  18 double distance_cosine(const Eigen::VectorXd& u, const Eigen::VectorXd& v);
  19
  20 // Computes  the Euclidean distance between u and v
  21 // https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.distance.euclidean.html#scipy.spatial.distance.euclidean
  22 double distance_euclidean(const Eigen::VectorXd& u, const Eigen::VectorXd& v);
  23
  24 #endif // _HUNGARIAN_H_