提取有角度的形狀的最小矩形,求其旋轉角度、長寬



這個程序搞了許久

#include <iostream>
#include <vector>
#include <opencv2/opencv.hpp> 
#include <opencv2/core/core.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>


#pragma comment(lib,"opencv_core249d.lib")      
#pragma comment(lib,"opencv_highgui249d.lib")      
#pragma comment(lib,"opencv_imgproc249d.lib") 

#define PI 3.1415926

using namespace std;
using namespace cv;


int main()
{
	// Read input binary image
	RNG rng(12345);
	char *image_name = "P2.png";
	Mat image = imread(image_name, 0);
	if (!image.data)
		return 0;

	namedWindow("Binary Image");
	imshow("Binary Image", image);

	// 從文件中加載原圖  
	Mat pSrcImage = imread(image_name, CV_LOAD_IMAGE_UNCHANGED);
	//IplImage *pSrcImage = cvLoadImage("D2.bmp", CV_LOAD_IMAGE_UNCHANGED);

	// 轉爲2值圖
	threshold(pSrcImage, pSrcImage, 200, 255, cv::THRESH_BINARY_INV);
	//cvThreshold(pSrcImage, pSrcImage, 200, 255, cv::THRESH_BINARY_INV);

	imwrite("binary.jpg", image);

	vector<std::vector<cv::Point>> contours;
	Mat canny_output;
	Canny(pSrcImage, canny_output, 80, 255, 3);
	vector<Vec4i> hierarchy;
	findContours(canny_output, contours, hierarchy, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_SIMPLE, Point(0, 0));

	/// 對每個找到的輪廓創建可傾斜的邊界框和橢圓
	vector<RotatedRect> minRect(contours.size());
	vector<RotatedRect> minEllipse(contours.size());
	for (int i = 0; i < contours.size(); i++)
	{
		minRect[i] = minAreaRect(Mat(contours[i]));
		if (contours[i].size() > 5)
		{
			minEllipse[i] = fitEllipse(Mat(contours[i]));
		}
	}
	/// 繪出輪廓及其可傾斜的邊界框和邊界橢圓

	CvBox2D     End_Rage2D;
	CvPoint2D32f rectpoint[4];
	Point2f rect_points[4];
	
	Mat drawing = Mat::zeros(canny_output.size(), CV_8UC3);
	for (int i = 0; i< contours.size(); i++)
	{
		//End_Rage2D = cvMinAreaRect2(contour);
		End_Rage2D = minAreaRect(Mat(contours[i]));
		Scalar color = Scalar(rng.uniform(0, 255), rng.uniform(0, 255), rng.uniform(0, 255));
		// contour
		drawContours(drawing, contours, i, color, 1, 8, vector<Vec4i>(), 0, Point());
		// ellipse  橢圓
		//ellipse(drawing, minEllipse[i], color, 2, 8);
		// rotated rectangle
		Point2f rect_points[4]; minRect[i].points(rect_points);
		for (int j = 0; j < 4; j++)
		{
         line(drawing, rect_points[j], rect_points[(j + 1) % 4], color, 1, 8);
		 cout << "座標點:"<<rect_points[j] << "........" << rect_points[(j + 1) % 4] << endl;
		}
		
		
			int aaa = 0, bbb = 0, k = 0;
			aaa = (int)sqrt((pow((rect_points[0].x - rect_points[1].x), 2) + pow((rect_points[0].y - rect_points[1].y), 2)));
			bbb = (int)sqrt((pow((rect_points[0].x - rect_points[3].x), 2) + pow((rect_points[0].y - rect_points[3].y), 2)));

			if (aaa<bbb)
			{
				k = aaa;
				aaa = bbb;
				bbb = k;
			}
			printf("最小外接矩形的長爲:%d,寬爲:%d。\n\n", aaa, bbb);
		    cout << " angle:\n" << (float)End_Rage2D.angle << std::endl;      //被測物體旋轉角度 
	}    

	namedWindow("Contours", CV_WINDOW_AUTOSIZE);
	imshow("Contours1", drawing);



	////以上成功提取斜框/////

	waitKey(0);
	return 0;

}





發佈了36 篇原創文章 · 獲贊 17 · 訪問量 6萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章