Hu Chunming / vpt_proj

HungarianAlgorithm.h 615 Bytes
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 
#ifndef HUNGARIANALGORITHM_H_
#define HUNGARIANALGORITHM_H_

#include <vector>
#ifdef _MSC_VER
        #include <cv.h>
#else
        #include <opencv2/core/core.hpp>
        #include <opencv2/highgui/highgui.hpp>
        #include <opencv2/imgproc/imgproc.hpp>
#endif
#include <assert.h>
#include <iostream>

using namespace cv;
using namespace std;



void  extractGrids(const Mat &A, const vector<bool> &extractRows, const vector<bool> &extractCols, Mat &B);
void  extractDots(const Mat &A, const vector<bool> &extract, Mat &B);
void  munkres(Mat &IoUMat, vector<int> &assignment);

#endif