Zou XiKun / left_over

HungarianAlgorithm.h 564 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 std;



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

#endif