Function testHomographyFitting

examples/cpp_example.cpp:698–875

}

void testHomographyFitting(
	double ransac_confidence_, // The confidence required
	double maximum_threshold_, // The maximum threshold value
	std::string test_scene_, // The name of the current test scene
	bool use_magsac_plus_plus_, // A flag to decide if MAGSAC++ or MAGSAC should be used
	bool draw_results_, // A flag determining if the results should be visualized
	double drawing_threshold_) // The threshold used for visualizing the results
{
	// Print the name of the current test scene
	std::cout  << "Processed scene = '" << test_scene_ << "'.";

	// Load the images of the current test scene
	cv::Mat image1 = cv::imread("../data/homography/" + test_scene_ + "A.png");
	cv::Mat image2 = cv::imread("../data/homography/" + test_scene_ + "B.png");
	if (image1.cols == 0 || image2.cols == 0) // If the images have not been loaded, try to load them as jpg files.
	{
		image1 = cv::imread("../data/homography/" + test_scene_ + "A.jpg");
		image2 = cv::imread("../data/homography/" + test_scene_ + "B.jpg");
	}

	// If the images have not been loaded, return
	if (image1.cols == 0 || image2.cols == 0)
	{
		std::cerr << "A problem occured when loading the images for test scene " << test_scene_ << ".";
		return;
	}

	cv::Mat points; // The point correspondences, each is of format "x1 y1 x2 y2"
	std::vector<int> ground_truth_labels; // The ground truth labeling provided in the dataset

	// A function loading the points from files
	readAnnotatedPoints("../data/homography/" + test_scene_ + "_pts.txt", // The path where the reference labeling and the points are found
		points, // All data points 
		ground_truth_labels); // The reference labeling

	// The number of points in the datasets
	const size_t point_number = points.rows; // The number of points in the scene

	if (point_number == 0) // If there are no points, return
	{
		std::cerr << "A problem occured when loading the annotated points for test scene " << test_scene_ << ".";
		return;
	}

	magsac::utils::DefaultHomographyEstimator estimator; // The robust homography estimator class containing the function for the fitting and residual calculation
	gcransac::Homography model; // The estimated model

	// In this used datasets, the manually selected inliers are not all inliers but a subset of them.
	// Therefore, the manually selected inliers are augmented as follows: 
	// (i) First, the implied model is estimated from the manually selected inliers.
	// (ii) Second, the inliers of the ground truth model are selected.
	std::vector<int> refined_labels = ground_truth_labels;
	refineManualLabeling<gcransac::Homography, magsac::utils::DefaultHomographyEstimator>(
		points, // The data points
		refined_labels, // The refined labeling
		estimator, // The model estimator
		2.0); // The used threshold in pixels