devbox
运维做开发
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

ROI 接口便捷修改

作者：运维做开发 | 2024-07-14 16:42:08

赞

踩

ROI 接口便捷修改

传入的图片截取ROI后再进入识别接口

（识别接口比ROI接口的函数参数少一个传入的ROI）

无点只有点集

 返回双点集

  1. //平直冷侧翅片
  2. bool ImageProcessingTest::straightColdSideFin_ROI(cv::Mat img, cv::Rect ROI, std::vector<cv::Point>& topList, std::vector<cv::Point>& bottomList, cv::Mat & canvas, bool debug)
  3. {
  4. 	int endRes = 0;
  5. 	bool result = false;
  6. 	img.copyTo(canvas);
  7. 	//top = cv::Point(0, 0);
  8. 	//bottom = cv::Point(0, 0);
  9.  
  10. 	cv::Mat imgOriginal;
  11. 	bool ROI_flag;//是否有给信息区
  12. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  13. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  14. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  15. 	{
  16. 		img.copyTo(imgOriginal);
  17. 		ROI_flag = false;
  18. 	}
  19. 	else
  20. 	{
  21. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  22. 		{
  23. 			printf("信息区两点疑似传反！");
  24. 			return false;
  25. 		}
  26. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  27. 		{
  28. 			printf("信息区的框选超出图像范围！");
  29. 			return false;
  30. 		}
  31. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  32. 		{
  33. 			printf("信息区大小超出图像大小！");
  34. 			return false;
  35. 		}
  36.  
  37. 		//此时不合法的ROI都已提前返回false
  38.  
  39. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  40. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  41. 		ROI_flag = true;
  42. 	}
  43.  
  44. 	result = straightColdSideFin(imgOriginal, topList, bottomList, canvas, debug);//调用识别算法
  45.  
  46. 	if (debug) {
  47. 		for (int i = 0; i < topList.size(); i++) {
  48. 			cv::Point top = topList.at(i);
  49. 			top += ROI_tl;
  50. 			if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】\n", top.x, top.y);
  51. 		}
  52. 		for (int i = 0; i < bottomList.size(); i++) {
  53. 			cv::Point bottom = bottomList.at(i);
  54. 			bottom += ROI_tl;
  55. 			if (debug) printf("-------------------------------------****------ roiF  bottom=【%d, %d】\n", bottom.x, bottom.y);
  56. 		}
  57. 	}
  58. 	//if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  59. 	//img.copyTo(canvas);
  60.  
  61.  
  62. 	x值和y值补上ROI左上点x和y
  63. 	//if (ROI_flag) {
  64. 	//	if (Corner != cv::Point(0, 0))
  65. 	//	{
  66. 	//		Corner.x += ROI_tl.x;
  67. 	//		Corner.y += ROI_tl.y;
  68. 	//	}
  69. 	//}
  70.  
  71. 	img.copyTo(canvas);//调试时注释掉
  72. 	if (result)
  73. 	{
  74. 		cv::Point textP(6, 0);//演示用
  75. 		endRes = 1;
  76. 		for (int i = 0; i < topList.size(); i++) {
  77. 			topList.at(i) += ROI_tl;
  78. 			cv::Point top = topList.at(i);
  79. 			circle(canvas, top, 2, cv::Scalar(0, 69, 255), -1);
  80. 			cv::putText(canvas, to_string(i), top + textP, cv::FONT_HERSHEY_COMPLEX, 0.45, cv::Scalar(0, 69, 255), 1);//演示用
  81. 		}
  82. 		for (int i = 0; i < bottomList.size(); i++) {
  83. 			bottomList.at(i) += ROI_tl;
  84. 			cv::Point bottom = bottomList.at(i);
  85. 			circle(canvas, bottom, 2, cv::Scalar(255, 69, 0), -1);
  86. 			cv::putText(canvas, to_string(i), bottom + textP, cv::FONT_HERSHEY_COMPLEX, 0.45, cv::Scalar(255, 69, 0), 1);//演示用
  87. 		}
  88. 		printf("topList.size()=%d , bottomList.size()=%d \n", topList.size(), bottomList.size());
  89. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  90. 	}
  91. 	else
  92. 	{
  93. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  94. 	}
  95.  
  96.  
  97. 	cv::Mat ROIcanvas;
  98. 	img.copyTo(ROIcanvas);
  99. 	if (debug) {
  100. 		if (debug) printf("topList.size()=%d , bottomList.size()=%d \n", topList.size(), bottomList.size());
  101. 		for (int i = 0; i < topList.size(); i++) {
  102. 			cv::Point top = topList.at(i);
  103. 			circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  104. 			if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】\n", top.x, top.y);
  105. 		}
  106. 		for (int i = 0; i < bottomList.size(); i++) {
  107. 			cv::Point bottom = bottomList.at(i);
  108. 			circle(ROIcanvas, bottom, 3, cv::Scalar(255, 69, 0), 1);
  109. 			if (debug) printf("-------------------------------------****------ roiF  bottom=【%d, %d】\n", bottom.x, bottom.y);
  110. 		}
  111. 	}
  112. 	//if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  113. 	//if (debug) circle(ROIcanvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  114. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  115. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  116. 	//if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  117. 	if (debug) imshow("ROI", ROIcanvas);
  118. 	if (debug) imshow("ROI接口_canvas", canvas);
  119. 	//ROIcanvas.copyTo(canvas);
  120.  
  121. 	return result;
  122. }

调用main：

  1. //平直冷侧翅片：straightColdSideFin_ROI
  2. int main333() {
  3. 	bool flag = false;
  4. 	std::string filePath = "../img/straightColdSideFin/0616img";
  5.  
  6.  
  7. 	string resPath = "/output";
  8. 	//string resPath = filePath + "./output";
  9.  
  10. 	
  11. 	cv::Point tlP = cv::Point(200, 0);
  12. 	cv::Point brP = cv::Point(950, 1024);
  13. 	std::vector<cv::Point> topList;
  14. 	std::vector<cv::Point> bottomList;
  15. 	cv::Mat canvas;
  16. 	ImageProcessingTest m_ImageProcessing;
  17. 	ofstream ofs("log.txt", ios::out);
  18. 	if (ofs)
  19. 	{
  20. 		ofs.clear();
  21. 		ofs << "日志启动" << endl;
  22. 	}
  23.  
  24. 	vector<string> path_name;
  25.  
  26.  
  27. 	bool   lianxu = false; //循环测试
  28.  
  29. 	if (lianxu)
  30. 	{
  31. 		string path = filePath + cv::format("\\Image_20230614160744823.png");
  32. 		//string path = filePath + cv::format("\\Image_20230609111800152.png");
  33. 		//string path = filePath + cv::format("\\Image_20230601143638388.png");
  34. 		//string path = filePath + cv::format("\\Image_20230601143505168.png");
  35. 		//string path = filePath + cv::format("\\Image_20230609113049539.png");
  36. 		cv::Mat src = cv::imread(path);
  37. 		src.copyTo(canvas);
  38. 		cv::namedWindow("main_OriginalIMG", cv::RECURS_FILTER);
  39. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  40. 		cv::imshow("main_OriginalIMG", src);
  41.  
  42. 		//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  43. 		cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  44.  
  45.  
  46. 		bool flag = m_ImageProcessing.straightColdSideFin_ROI(src, ROI, topList, bottomList, canvas, true);
  47.  
  48. 		cout << "\nmain ==== flag " << flag << endl;
  49. 		if (flag) {
  50. 			for (int i = 0; i < topList.size(); i++) {
  51. 				cv::Point top = topList.at(i);
  52. 				cout << "top = " << top.x << " " << top.y << endl;
  53. 				circle(canvas, top, 6, cv::Scalar(0, 69, 255), 1);
  54. 			}
  55. 			for (int i = 0; i < bottomList.size(); i++) {
  56. 				cv::Point bottom = bottomList.at(i);
  57. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  58. 				circle(canvas, bottom, 6, cv::Scalar(255, 69, 0), 1);
  59. 			}
  60. 			printf("topList.size()=%d , bottomList.size()=%d \n", topList.size(), bottomList.size());
  61. 		}
  62. 		canvas.copyTo(src);
  63. 		//cout << "\nmain ==== i " << endl;
  64. 		//cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  65. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  66. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  67. 		cv::imshow("main-结果图src", src);
  68. 	}
  69. 	else
  70. 	{
  71. 		vector<string> path_name;
  72. 		getAllFiles(filePath, path_name);
  73. 		cout << "path_name.size()=" << path_name.size() << endl;
  74.  
  75. 		int number = 1;
  76. 		for (auto i : path_name)
  77. 		{
  78. 			cv::Mat src = cv::imread(filePath + "\\" + i);
  79. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  80. 			src.copyTo(canvas);
  81. 			cv::imshow("main_OriginalIMG", src);
  82.  
  83. 			//将结果图写入文件方便查看结果
  84. 			int len = filePath.length();
  85. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  86. 			cout << "i=" << i << endl;
  87. 			int time_start = clock();
  88. 			int time_end = clock();
  89.  
  90. 			//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  91. 			cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  92.  
  93.  
  94. 			bool flag = m_ImageProcessing.straightColdSideFin_ROI(src, ROI, topList, bottomList, canvas);
  95.  
  96.  
  97. 			time_end = clock();
  98. 			printf("\n耗时 %d ms \n", time_end - time_start);
  99. 			cout << "\nmain ==== flag " << flag << endl;
  100. 			if (flag) {
  101. 				for (int i = 0; i < topList.size(); i++) {
  102. 					cv::Point top = topList.at(i);
  103. 					cout << "top = " << top.x << " " << top.y << endl;
  104. 				}
  105. 				for (int i = 0; i < bottomList.size(); i++) {
  106. 					cv::Point bottom = bottomList.at(i);
  107. 					cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  108. 				}
  109. 				printf("topList.size()=%d , bottomList.size()=%d \n", topList.size(), bottomList.size());
  110. 			}
  111. 			canvas.copyTo(src);
  112. 			//cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  113. 			cv::imshow("main-结果图src", src);
  114. 			cv::waitKey(10);
  115.  
  116.  
  117. 			//将结果图写入文件方便查看结果
  118. 			char*  filePath_ = new char[len + 1];
  119. 			strcpy(filePath_, filePath.c_str());
  120.  
  121. 			len = filePath.length();
  122. 			char*  resPath_ = new char[len + 1];
  123. 			strcpy(resPath_, resPath.c_str());
  124. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  125.  
  126. 			string out = filePath + resPath;
  127. 			cout << "\nmain ==== i " << i << endl;
  128. 			out.append("\\").append(i);
  129.  
  130. 			cout << "number【" << number << "】res path=" << out << endl;
  131. 			cv::imwrite(out, src);
  132. 			number++;
  133.  
  134. 			cout << "path_name.size() = " << path_name.size() << endl;
  135. 			cout << "*********************************************************************************************************************" << endl;
  136. 			if (number == path_name.size()) break;
  137. 		}
  138. 	}
  139. 	cout << "main 循环结束！！！" << endl;
  140.  
  141. 	ofs.close();
  142. 	cv::waitKey(0);
  143. 	system("Pause");
  144. 	return 0;
  145. }

单点

返回只有单点 

  1. bool ImageProcess::arcPlusLine_ROI(cv::Mat img, cv::Rect ROI, cv::Point & top, cv::Mat & canvas, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7.  
  8. 	cv::Mat imgOriginal;
  9. 	bool ROI_flag;//是否有给信息区
  10. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  11. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  12. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  13. 	{
  14. 		img.copyTo(imgOriginal);
  15. 		ROI_flag = false;
  16. 	}
  17. 	else
  18. 	{
  19. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  20. 		{
  21. 			printf("信息区两点疑似传反！");
  22. 			return false;
  23. 		}
  24. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  25. 		{
  26. 			printf("信息区的框选超出图像范围！");
  27. 			return false;
  28. 		}
  29. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  30. 		{
  31. 			printf("信息区大小超出图像大小！");
  32. 			return false;
  33. 		}
  34.  
  35. 		//此时不合法的ROI都已提前返回false
  36.  
  37. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  38. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  39. 		ROI_flag = true;
  40. 	}
  41.  
  42. 	result = arcPlusLine(imgOriginal, top, canvas, debug);//调用识别算法
  43.  
  44. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】\n", top.x, top.y);
  45. 	//img.copyTo(canvas);
  46.  
  47.  
  48. 	//img.copyTo(canvas);//调试时注释掉
  49.  
  50. 	cv::Mat t1;
  51. 	//canvas.convertTo(t1, -1, 900 / 100.0, 200 - 100);//第一次线性变换100
  52. 	canvas.convertTo(t1, -1, 900 / 100.0, 200 - 100);//第一次线性变换100
  53. 	if (debug)cv::imshow("线性变换ROI", t1);
  54. 	t1.copyTo(canvas);
  55.  
  56. 	if (result)
  57. 	{
  58. 		endRes = 1;
  59. 		top += ROI_tl;
  60. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  61. 		int w = 38;
  62. 		int thickness = 4;
  63. 		cv::Point curPoint = top;
  64. 		cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(235, 135, 40), thickness);
  65. 		circle(canvas, top, 2, cv::Scalar(0, 69, 255), -1);
  66. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  67. 	}
  68. 	else
  69. 	{
  70. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  71. 	}
  72.  
  73.  
  74. 	cv::Mat ROIcanvas;
  75. 	img.copyTo(ROIcanvas);
  76. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  77. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  78. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  79. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  80. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】 \n", top.x, top.y);
  81. 	if (debug) imshow("ROI", ROIcanvas);
  82. 	if (debug) imshow("ROI_canvas", canvas);
  83. 	//ROIcanvas.copyTo(canvas);
  84.  
  85.  
  86. 	//cv::Rect roi_(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  87. 	//if(!ROI_flag) canvas.copyTo(canvas);
  88. 	//else canvas = canvas(roi_);	//裁剪出的ROI区域
  89.  
  90. 	return result;
  91. }

调用main：

  1. //圆弧+直线 内角：arcPlusLine_ROI
  2. int main/*arcl*/() {
  3. 	bool flag = false;
  4. 	//std::string filePath = "E://vsproject//WeldingLine//1219img//in//宁波金波";
  5. 	//std::string filePath = "E://vsproject//WeldingLine//1219img//in//宁波金波//down";
  6. 	std::string filePath = "E://vsproject//WeldingLine//1219img//in//宁波金波//LogImg";
  7. 	//std::string filePath = "../img/ThreadedRoundPipe/0530img";
  8.  
  9.  
  10. 	string resPath = "/output";
  11. 	//string resPath = filePath + "./output";
  12.  
  13.  
  14. 	cv::Point tlP = cv::Point(200, 340);
  15. 	cv::Point brP = cv::Point(950, 999);
  16. 	cv::Point top(90, 90);
  17. 	cv::Mat canvas;
  18. 	VisualInterface m_ImageProcessing;
  19. 	//ImageProcess m_ImageProcessing;
  20. 	ofstream ofs("log.txt", ios::out);
  21. 	if (ofs)
  22. 	{
  23. 		ofs.clear();
  24. 		ofs << "日志启动" << endl;
  25. 	}
  26.  
  27. 	vector<string> path_name;
  28. 	bool lineOnTop = true;
  29.  
  30.  
  31. 	bool   getROI = !true;
  32. 	bool   lianxu = false; //循环测试
  33.  
  34. 	if (!lianxu)
  35. 	{
  36. 		string name = cv::format("\\Img_2024_03_12_15_01_57_350_TYPE_12_ROI_486_64_257_474_src.png");
  37. 		string path = filePath + name;
  38. 		//string path = filePath + cv::format("\\Image_20240228115544906.bmp");
  39. 		cv::Mat src = cv::imread(path);
  40. 		src.copyTo(canvas);
  41. 		cv::namedWindow("main_OriginalIMG", cv::RECURS_FILTER);
  42. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  43. 		cv::imshow("main_OriginalIMG", src);
  44.  
  45. 		cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  46. 		//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  47. 		if (getROI) ROI = getROIFromString(name);
  48.  
  49.  
  50. 		bool flag = m_ImageProcessing.arcPlusLine_ROI(src, ROI, top, canvas, lineOnTop, true);
  51.  
  52. 		cout << "\nmain ==== flag " << flag << endl;
  53. 		if (flag) {
  54. 			cout << "top = " << top.x << " " << top.y << endl;
  55. 		}
  56. 		canvas.copyTo(src);
  57. 		//cout << "\nmain ==== i " << endl;
  58. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  59. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 165), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  60. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  61. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  62. 		cv::imshow("main-结果图src", src);
  63. 	}
  64. 	else
  65. 	{
  66. 		vector<string> path_name;
  67. 		getAllFiles(filePath, path_name);
  68. 		cout << "path_name.size()=" << path_name.size() << endl;
  69.  
  70. 		int number = 1;
  71. 		for (auto i : path_name)
  72. 		{
  73. 			if (i.find("_res") != std::string::npos ) continue;
  74. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  75. 			cv::Mat src = cv::imread(filePath + "\\" + i);
  76. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  77. 			src.copyTo(canvas);
  78. 			cv::imshow("main_OriginalIMG", src);
  79.  
  80. 			//将结果图写入文件方便查看结果
  81. 			int len = filePath.length();
  82. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  83. 			cout << "i=" << i << endl;
  84. 			int time_start = clock();
  85. 			int time_end = clock();
  86.  
  87. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  88. 			//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  89. 			if (getROI) ROI = getROIFromString(i);
  90.  
  91.  
  92. 			bool flag = m_ImageProcessing.arcPlusLine_ROI(src, ROI, top, canvas, lineOnTop, false);
  93.  
  94.  
  95. 			time_end = clock();
  96. 			printf("\n耗时 %d ms \n", time_end - time_start);
  97. 			cout << "\nmain ==== flag " << flag << endl;
  98. 			if (flag) {
  99. 				cout << "top = " << top.x << " " << top.y << endl;
  100. 			}
  101. 			canvas.copyTo(src);
  102. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  103. 			cv::imshow("main-结果图src", src);
  104. 			cv::waitKey(10);
  105.  
  106.  
  107. 			//将结果图写入文件方便查看结果
  108. 			char*  filePath_ = new char[len + 1];
  109. 			strcpy(filePath_, filePath.c_str());
  110.  
  111. 			len = filePath.length();
  112. 			char*  resPath_ = new char[len + 1];
  113. 			strcpy(resPath_, resPath.c_str());
  114. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  115.  
  116. 			string out = filePath + resPath;
  117. 			cout << "\nmain ==== i " << i << endl;
  118. 			out.append("\\").append(i);
  119.  
  120. 			cout << "number【" << number << "】res path=" << out << endl;
  121. 			cv::imwrite(out, src);
  122. 			number++;
  123.  
  124. 			cout << "path_name.size() = " << path_name.size() << endl;
  125. 			cout << "*********************************************************************************************************************" << endl;
  126. 			if (number == path_name.size()) break;
  127. 		}
  128. 	}
  129. 	cout << "main 循环结束！！！" << endl;
  130.  
  131. 	ofs.close();
  132. 	cv::waitKey(0);
  133. 	system("Pause");
  134. 	return 0;
  135. }
  1. cv::Rect getROIFromString(std::string str) {
  2. 	//图像名字格式：“SrcImg_2023_10_11_11_39_19_002_ROI_600_396_137_218_src.png”
  3. 	if (str.empty()) return cv::Rect(0, 0, 0, 0);
  4. 	using namespace std;
  5.  
  6. 	const char *split = "_";
  7. 	char *p = strtok((char*)str.c_str(), split);
  8. 	std::vector<char*>data;
  9. 	while (p != NULL) {
  10. 		cout << p << endl;
  11. 		data.push_back(p);
  12. 		p = strtok(NULL, split);
  13. 	}
  14. 	//9 12;
  15.  
  16. 	if (data.size() < 12)return cv::Rect();
  17.  
  18. 	return cv::Rect(
  19. 		atoi(data[data.size() - 5]),
  20. 		atoi(data[data.size() - 4]),
  21. 		atoi(data[data.size() - 3]),
  22. 		atoi(data[data.size() - 2]));
  23.  
  24. 	//return cv::Rect(
  25. 	//	atoi(data[9]),
  26. 	//	atoi(data[10]),
  27. 	//	atoi(data[11]),
  28. 	//	atoi(data[12]));
  29. }

 返回只有单点，传入(高/低曝光图)两张原图，返回单点(高曝光图和激光图结合时使用)

  1. int ImageProcess::getPnt_2Dwith3D_MaiSi(cv::Mat imgHE, cv::Mat imgLE, cv::Mat & canvas, cv::Point & resPoint, bool &MaiSi_isTopPnt, cv::Rect ROI, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	imgHE.copyTo(canvas);
  6. 	cv::addWeighted(imgHE, 0.5, imgLE, 0.5, 0, canvas);
  7. 	ImageProcess pd;
  8.  
  9. 	cv::Mat imgHE_roi;
  10. 	cv::Mat imgOriginal;
  11. 	bool ROI_flag;//是否有给信息区
  12. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  13. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  14. 	cv::Mat img;
  15. 	imgLE.copyTo(img);
  16. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0))
  17. 	{
  18. 		img.copyTo(imgOriginal);
  19. 		imgHE.copyTo(imgHE_roi);
  20. 		ROI_flag = false;
  21. 	}
  22. 	else
  23. 	{
  24. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  25. 		{
  26. 			printf("信息区两点疑似传反！");
  27. 			return false;
  28. 		}
  29. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  30. 		{
  31. 			printf("信息区的框选超出图像范围！");
  32. 			return false;
  33. 		}
  34. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  35. 		{
  36. 			printf("信息区大小超出图像大小！");
  37. 			return false;
  38. 		}
  39.  
  40. 		//此时不合法的ROI都已提前返回false
  41.  
  42. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  43. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  44. 		imgHE_roi = imgHE(roi);
  45. 		ROI_flag = true;
  46. 		if (debug) {
  47. 			cv::Mat tmp;
  48. 			img.copyTo(tmp);
  49. 			rectangle(tmp, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  50. 			if (debug) cv::imshow("根据roi裁剪后进入识别的图", tmp);
  51. 		}
  52. 	}
  53.  
  54.  
  55. 	result = getPnt_2Dwith3D_MaiSi(imgHE_roi, imgOriginal, canvas, resPoint, MaiSi_isTopPnt, debug);
  56.  
  57.  
  58.  
  59. 	cv::addWeighted(imgHE, 0.5, imgLE, 0.5, 0, canvas);
  60. 	//x值和y值补上ROI左上点x和y
  61. 	if (ROI_flag)
  62. 	{
  63. 		if (resPoint != cv::Point(0, 0))
  64. 		{
  65. 			resPoint.x += ROI_tl.x;
  66. 			resPoint.y += ROI_tl.y;
  67. 		}
  68. 	}
  69. 	if (debug) std::cout << img.size() << imgOriginal.size() << std::endl;
  70.  
  71. 	if (result)
  72. 	{
  73. 		endRes = 1;
  74. 		cv::circle(canvas, resPoint, 3, cv::Scalar(0, 0, 255), -1, 8, 0);
  75. 		putText(canvas, "true", { cv::Point(250, 250) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  76. 		//putText(canvas, "EmbWare_Leg", { cv::Point(250, 350) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  77. 		if (MaiSi_isTopPnt) {
  78. 			putText(canvas, "isTopPnt", { resPoint + cv::Point(15, 5) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(195, 165, 255), 1);
  79. 		}
  80. 		else {
  81. 			putText(canvas, "isBotPnt", { resPoint + cv::Point(15, 5) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(155, 155, 255), 1);
  82. 		}
  83. 		Mark(canvas);
  84. 	}
  85. 	else
  86. 	{
  87. 		putText(canvas, "false", { cv::Point(250, 250) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  88. 		Mark(canvas);
  89. 	}
  90. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  91.  
  92. 	return endRes;
  93. }

返回单点+上下翻转再进入识别

  1. 	/*
  2. 	//圆弧+直线  求内角
  3. 	cv::Mat img, 原图
  4. 	cv::Rect ROI,
  5. 	cv::Point& Corner, 输出角点
  6. 	cv::Mat &canvas,效果图可视化
  7. 	bool lineOnTop = true,直线部分是否为较上端的角边，true则为是，false反之
  8. 	*/
  9. 	bool arcPlusLine(cv::Mat imgOriginal, cv::Point& Corner, cv::Mat &canvas, bool debug = false);
  10. 	bool arcPlusLine_ROI(cv::Mat img, cv::Rect ROI, cv::Point& Corner, cv::Mat &canvas, bool lineOnTop = true, bool debug = false);
  1. bool ImageProcess::arcPlusLine_ROI(cv::Mat img, cv::Rect ROI, cv::Point & top, cv::Mat & canvas, bool lineOnTop, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7.  
  8. 	cv::Mat imgOriginal;
  9. 	bool ROI_flag;//是否有给信息区
  10. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  11. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  12. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  13. 	{
  14. 		img.copyTo(imgOriginal);
  15. 		ROI_flag = false;
  16. 	}
  17. 	else
  18. 	{
  19. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  20. 		{
  21. 			printf("信息区两点疑似传反！");
  22. 			return false;
  23. 		}
  24. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  25. 		{
  26. 			printf("信息区的框选超出图像范围！");
  27. 			return false;
  28. 		}
  29. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  30. 		{
  31. 			printf("信息区大小超出图像大小！");
  32. 			return false;
  33. 		}
  34.  
  35. 		//此时不合法的ROI都已提前返回false
  36.  
  37. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  38. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  39. 		ROI_flag = true;
  40. 	}
  41.  
  42. 	if (lineOnTop) {
  43. 		result = arcPlusLine(imgOriginal, top, canvas, debug);//调用识别算法
  44. 	}
  45. 	else {
  46. 		cv::Mat fpMt;
  47. 		cv::flip(imgOriginal, fpMt, 0);		// 上下翻转 x对称
  48. 		result = arcPlusLine(fpMt, top, canvas, debug);//调用识别算法
  49. 		if (result) {
  50. 			top.y = fpMt.rows - top.y - 1;
  51. 		}
  52. 	}
  53. 	cv::Mat lockCanvas = canvas.clone();
  54.  
  55.  
  56. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】\n", top.x, top.y);
  57. 	//img.copyTo(canvas);
  58.  
  59.  
  60. 	//img.copyTo(canvas);//调试时注释掉
  61.  
  62. 	//cv::Mat t1;
  63. 	//canvas.convertTo(t1, -1, 900 / 100.0, 200 - 100);//第一次线性变换100
  64. 	//if (debug)cv::imshow("线性变换ROI", t1);
  65. 	//t1.copyTo(canvas);
  66.  
  67. 	if (result)
  68. 	{
  69. 		endRes = 1;
  70. 		top += ROI_tl;
  71. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  72. 		int w = 38;
  73. 		int thickness = 4;
  74. 		cv::Point curPoint = top;
  75. 		cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(235, 135, 40), thickness);
  76. 		circle(canvas, top, 2, cv::Scalar(0, 69, 255), -1);
  77. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  78. 	}
  79. 	else
  80. 	{
  81. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  82. 	}
  83.  
  84.  
  85. 	cv::Mat ROIcanvas;
  86. 	img.copyTo(ROIcanvas);
  87. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  88. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  89. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  90. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  91. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】 \n", top.x, top.y);
  92. 	if (debug) imshow("ROI", ROIcanvas);
  93. 	if (debug) imshow("ROI_canvas", canvas);
  94. 	//ROIcanvas.copyTo(canvas);
  95.  
  96.  
  97. 	//cv::Rect roi_(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  98. 	//if(!ROI_flag) canvas.copyTo(canvas);
  99. 	//else canvas = canvas(roi_);	//裁剪出的ROI区域
  100.  
  101. 	lockCanvas.copyTo(canvas);
  102. 	return result;
  103. }

调用main：

  1. // 内角：arcPlusLine_ROI
  2. int main/*arcl*/() {
  3. 	bool flag = false;
  4. 	std::string filePath = "E://vsproject//WeldingLine//1219img//in//宁波金波";
  5. 	//std::string filePath = "E://vsproject//WeldingLine//1219img//in//宁波金波//down";
  6. 	//std::string filePath = "../img/ThreadedRoundPipe/0530img";
  7.  
  8.  
  9. 	string resPath = "/output";
  10. 	//string resPath = filePath + "./output";
  11.  
  12.  
  13. 	cv::Point tlP = cv::Point(200, 340);
  14. 	cv::Point brP = cv::Point(950, 999);
  15. 	cv::Point top(90, 90);
  16. 	cv::Mat canvas;
  17. 	//VisualInterface m_ImageProcessing;
  18. 	ImageProcess m_ImageProcessing;
  19. 	ofstream ofs("log.txt", ios::out);
  20. 	if (ofs)
  21. 	{
  22. 		ofs.clear();
  23. 		ofs << "日志启动" << endl;
  24. 	}
  25.  
  26. 	vector<string> path_name;
  27. 	bool lineOnTop = true;
  28.  
  29.  
  30. 	bool   lianxu = false; //循环测试
  31.  
  32. 	if (lianxu)
  33. 	{
  34. 		string path = filePath + cv::format("\\Image_20240304155500627.png");
  35. 		//string path = filePath + cv::format("\\Image_20240228115544906.bmp");
  36. 		cv::Mat src = cv::imread(path);
  37. 		src.copyTo(canvas);
  38. 		cv::namedWindow("main_OriginalIMG", cv::RECURS_FILTER);
  39. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  40. 		cv::imshow("main_OriginalIMG", src);
  41.  
  42. 		cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  43. 		//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  44.  
  45.  
  46. 		bool flag = m_ImageProcessing.arcPlusLine_ROI(src, ROI, top, canvas, lineOnTop, true);
  47.  
  48. 		cout << "\nmain ==== flag " << flag << endl;
  49. 		if (flag) {
  50. 			cout << "top = " << top.x << " " << top.y << endl;
  51. 		}
  52. 		canvas.copyTo(src);
  53. 		//cout << "\nmain ==== i " << endl;
  54. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  55. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 165), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  56. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  57. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  58. 		cv::imshow("main-结果图src", src);
  59. 	}
  60. 	else
  61. 	{
  62. 		vector<string> path_name;
  63. 		getAllFiles(filePath, path_name);
  64. 		cout << "path_name.size()=" << path_name.size() << endl;
  65.  
  66. 		int number = 1;
  67. 		for (auto i : path_name)
  68. 		{
  69. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  70. 			cv::Mat src = cv::imread(filePath + "\\" + i);
  71. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  72. 			src.copyTo(canvas);
  73. 			cv::imshow("main_OriginalIMG", src);
  74.  
  75. 			//将结果图写入文件方便查看结果
  76. 			int len = filePath.length();
  77. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  78. 			cout << "i=" << i << endl;
  79. 			int time_start = clock();
  80. 			int time_end = clock();
  81.  
  82. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  83. 			//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  84.  
  85.  
  86. 			bool flag = m_ImageProcessing.arcPlusLine_ROI(src, ROI, top, canvas, lineOnTop, false);
  87.  
  88.  
  89. 			time_end = clock();
  90. 			printf("\n耗时 %d ms \n", time_end - time_start);
  91. 			cout << "\nmain ==== flag " << flag << endl;
  92. 			if (flag) {
  93. 				cout << "top = " << top.x << " " << top.y << endl;
  94. 			}
  95. 			canvas.copyTo(src);
  96. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  97. 			cv::imshow("main-结果图src", src);
  98. 			cv::waitKey(10);
  99.  
  100.  
  101. 			//将结果图写入文件方便查看结果
  102. 			char*  filePath_ = new char[len + 1];
  103. 			strcpy(filePath_, filePath.c_str());
  104.  
  105. 			len = filePath.length();
  106. 			char*  resPath_ = new char[len + 1];
  107. 			strcpy(resPath_, resPath.c_str());
  108. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  109.  
  110. 			string out = filePath + resPath;
  111. 			cout << "\nmain ==== i " << i << endl;
  112. 			out.append("\\").append(i);
  113.  
  114. 			cout << "number【" << number << "】res path=" << out << endl;
  115. 			cv::imwrite(out, src);
  116. 			number++;
  117.  
  118. 			cout << "path_name.size() = " << path_name.size() << endl;
  119. 			cout << "*********************************************************************************************************************" << endl;
  120. 			if (number == path_name.size()) break;
  121. 		}
  122. 	}
  123. 	cout << "main 循环结束！！！" << endl;
  124.  
  125. 	ofs.close();
  126. 	cv::waitKey(0);
  127. 	system("Pause");
  128. 	return 0;
  129. }

返回单点+1个点集+上下翻转再进入识别 

  1. bool PTank_imgProcess::arcPlusLine_ROI(cv::Mat img, cv::Rect ROI, cv::Point & top, std::vector<cv::Point>& allPntList, cv::Mat & canvas, bool lineOnTop, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7.  
  8. 	cv::Mat imgOriginal;
  9. 	bool ROI_flag;//是否有给信息区
  10. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  11. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  12. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  13. 	{
  14. 		img.copyTo(imgOriginal);
  15. 		ROI_flag = false;
  16. 	}
  17. 	else
  18. 	{
  19. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  20. 		{
  21. 			printf("信息区两点疑似传反！");
  22. 			return false;
  23. 		}
  24. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  25. 		{
  26. 			printf("信息区的框选超出图像范围！");
  27. 			return false;
  28. 		}
  29. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  30. 		{
  31. 			printf("信息区大小超出图像大小！");
  32. 			return false;
  33. 		}
  34.  
  35. 		//此时不合法的ROI都已提前返回false
  36.  
  37. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  38. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  39. 		ROI_flag = true;
  40. 	}
  41.  
  42. 	cv::Mat lockCanvas;
  43. 	if (lineOnTop) {
  44. 		result = arcPlusLine(imgOriginal, top, allPntList, canvas, debug);//调用识别算法
  45. 		canvas.copyTo(lockCanvas);
  46. 		//for (int i = 0; i < allPntList.size(); i++) {
  47. 		//	cv::line(lockCanvas, allPntList.at(i), cv::Point(allPntList.at(i).x - 3, allPntList.at(i).y), cv::Scalar(55, 195, 40), 1);
  48. 		//}
  49. 	}
  50. 	else {
  51. 		cv::Mat fpMt;
  52. 		cv::flip(imgOriginal, fpMt, 0);		// 上下翻转 x对称
  53. 		result = arcPlusLine(fpMt, top, allPntList, canvas, debug);//调用识别算法
  54. 		canvas.copyTo(lockCanvas);
  55. 		if (result) {
  56. 			top.y = fpMt.rows - top.y - 1;
  57. 			for (int i = 0; i < allPntList.size(); i++) {
  58. 				//cv::line(lockCanvas, allPntList.at(i), cv::Point(allPntList.at(i).x - 3, allPntList.at(i).y), cv::Scalar(55, 195, 40), 1);
  59. 				allPntList.at(i).y = fpMt.rows - allPntList.at(i).y - 1;
  60. 			}
  61. 		}
  62. 	}
  63.  
  64.  
  65. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】\n", top.x, top.y);
  66. 	//img.copyTo(canvas);
  67.  
  68.  
  69. 	//img.copyTo(canvas);//调试时注释掉
  70.  
  71. 	//cv::Mat t1;
  72. 	//canvas.convertTo(t1, -1, 900 / 100.0, 200 - 100);//第一次线性变换100
  73. 	//if (debug)cv::imshow("线性变换ROI", t1);
  74. 	//t1.copyTo(canvas);
  75.  
  76. 	if (result)
  77. 	{
  78. 		img.copyTo(canvas);
  79. 		endRes = 1;
  80. 		top += ROI_tl;
  81. 		for (int i = 0; i < allPntList.size(); i++) {
  82. 			allPntList.at(i) += ROI_tl;
  83. 		}
  84. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  85. 		int w = 38;
  86. 		int thickness = 4;
  87. 		cv::Point curPoint = top;
  88. 		cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(235, 135, 40), thickness);
  89. 		circle(canvas, top, 2, cv::Scalar(0, 69, 255), -1);
  90. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  91. 		w = 3;
  92. 		for (int i = 0; i < allPntList.size(); i++) {
  93. 			cv::Point curPoint = allPntList.at(i);
  94. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(55, 195, 40), 1);
  95. 		}
  96. 	}
  97. 	else
  98. 	{
  99. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  100. 	}
  101.  
  102.  
  103. 	cv::Mat ROIcanvas;
  104. 	img.copyTo(ROIcanvas);
  105. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  106. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  107. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  108. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  109. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】 \n", top.x, top.y);
  110. 	if (debug) imshow("ROI", ROIcanvas);
  111. 	if (debug) imshow("ROI_canvas", canvas);
  112. 	//ROIcanvas.copyTo(canvas);
  113.  
  114.  
  115. 	//cv::Rect roi_(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  116. 	//if(!ROI_flag) canvas.copyTo(canvas);
  117. 	//else canvas = canvas(roi_);	//裁剪出的ROI区域
  118.  
  119. 	if (result) lockCanvas.copyTo(canvas);
  120. 	cv::putText(canvas, to_string(result), cv::Point(30, 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  121. 	return result;
  122. }

调用main：

  1. // 内角：arcPlusLine_ROI 类似金波(金波优化版)
  2. int main/*arcl*/() {
  3. 	bool flag = false;
  4. 	//std::string filePath = "E://vsproject//压力罐\\img\\类金波/top";
  5. 	std::string filePath = "E://vsproject//压力罐\\img\\类金波";
  6. 	//std::string filePath = "E://vsproject//压力罐\\img\\叶子";
  7. 	//std::string filePath = "../img/ThreadedRoundPipe/0530img";
  8.  
  9.  
  10. 	string resPath = "/output";
  11. 	//string resPath = filePath + "./output";
  12.  
  13.  
  14. 	cv::Point tlP = cv::Point(200, 340);
  15. 	cv::Point brP = cv::Point(950, 999);
  16. 	cv::Point top(90, 90);
  17. 	std::vector<cv::Point> allPntList;
  18. 	cv::Mat canvas;
  19. 	PTank_imgProcess m_ImageProcessing;
  20. 	//ImageProcess m_ImageProcessing;
  21. 	ofstream ofs("log.txt", ios::out);
  22. 	if (ofs)
  23. 	{
  24. 		ofs.clear();
  25. 		ofs << "日志启动" << endl;
  26. 	}
  27.  
  28. 	vector<string> path_name;
  29. 	bool lineOnTop = !true;
  30. 	
  31.  
  32. 	bool   getROI = true;
  33. 	bool   lianxu = false; //循环测试
  34.  
  35. 	if (!lianxu)
  36. 	{
  37. 		string name = cv::format("\\Image_20240627151014832.png");
  38. 		string path = filePath + name;
  39. 		cv::Mat src = cv::imread(path);
  40. 		src.copyTo(canvas);
  41. 		cv::namedWindow("main_OriginalIMG", cv::RECURS_FILTER);
  42. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  43. 		cv::imshow("main_OriginalIMG", src);
  44.  
  45. 		cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  46. 		//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  47. 		if (getROI) ROI = getROIFromString(name);
  48.  
  49.  
  50. 		bool flag = m_ImageProcessing.arcPlusLine_ROI(src, ROI, top, allPntList, canvas, lineOnTop, true);
  51.  
  52. 		cout << "\nmain ==== flag " << flag << endl;
  53. 		if (flag) {
  54. 			cout << "top = " << top.x << " " << top.y << endl;
  55. 		}
  56. 		canvas.copyTo(src);
  57. 		//cout << "\nmain ==== i " << endl;
  58. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  59. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 165), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  60. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  61. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  62. 		cv::imshow("main-结果图src", src);
  63. 	}
  64. 	else
  65. 	{
  66. 		vector<string> path_name;
  67. 		getAllFiles(filePath, path_name);
  68. 		cout << "path_name.size()=" << path_name.size() << endl;
  69.  
  70. 		int number = 1;
  71. 		for (auto i : path_name)
  72. 		{
  73. 			if (i.find("_res") != std::string::npos) continue;
  74. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  75. 			cv::Mat src = cv::imread(filePath + "\\" + i);
  76. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  77. 			src.copyTo(canvas);
  78. 			cv::imshow("main_OriginalIMG", src);
  79.  
  80. 			//将结果图写入文件方便查看结果
  81. 			int len = filePath.length();
  82. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  83. 			cout << "i=" << i << endl;
  84. 			int time_start = clock();
  85. 			int time_end = clock();
  86.  
  87. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  88. 			//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  89. 			if (getROI) ROI = getROIFromString(i);
  90.  
  91.  
  92. 			bool flag = m_ImageProcessing.arcPlusLine_ROI(src, ROI, top, allPntList, canvas, lineOnTop, false);
  93.  
  94.  
  95. 			time_end = clock();
  96. 			printf("\n耗时 %d ms \n", time_end - time_start);
  97. 			cout << "\nmain ==== flag " << flag << endl;
  98. 			if (flag) {
  99. 				cout << "top = " << top.x << " " << top.y << endl;
  100. 			}
  101. 			canvas.copyTo(src);
  102. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  103. 			cv::imshow("main-结果图src", src);
  104. 			cv::waitKey(10);
  105.  
  106.  
  107. 			//将结果图写入文件方便查看结果
  108. 			char*  filePath_ = new char[len + 1];
  109. 			strcpy(filePath_, filePath.c_str());
  110.  
  111. 			len = filePath.length();
  112. 			char*  resPath_ = new char[len + 1];
  113. 			strcpy(resPath_, resPath.c_str());
  114. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  115.  
  116. 			string out = filePath + resPath;
  117. 			cout << "\nmain ==== i " << i << endl;
  118. 			out.append("\\").append(i);
  119.  
  120. 			cout << "number【" << number << "】res path=" << out << endl;
  121. 			cv::imwrite(out, src);
  122. 			number++;
  123.  
  124. 			cout << "path_name.size() = " << path_name.size() << endl;
  125. 			cout << "*********************************************************************************************************************" << endl;
  126. 			if (number == path_name.size()) break;
  127. 		}
  128. 	}
  129. 	cout << "main 循环结束！！！" << endl;
  130.  
  131. 	ofs.close();
  132. 	cv::waitKey(0);
  133. 	system("Pause");
  134. 	return 0;
  135. }

双点

返回只有双点 

  1. bool ImageProcessingTest::ThreadedRoundPipe_ROI(cv::Mat img, cv::Rect ROI, cv::Point& top, cv::Point& bottom, cv::Mat & canvas, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7. 	bottom = cv::Point(0, 0);
  8.  
  9. 	cv::Mat imgOriginal;
  10. 	bool ROI_flag;//是否有给信息区
  11. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  12. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  13. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  14. 	{
  15. 		img.copyTo(imgOriginal);
  16. 		ROI_flag = false;
  17. 	}
  18. 	else
  19. 	{
  20. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  21. 		{
  22. 			printf("信息区两点疑似传反！");
  23. 			return false;
  24. 		}
  25. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  26. 		{
  27. 			printf("信息区的框选超出图像范围！");
  28. 			return false;
  29. 		}
  30. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  31. 		{
  32. 			printf("信息区大小超出图像大小！");
  33. 			return false;
  34. 		}
  35.  
  36. 		//此时不合法的ROI都已提前返回false
  37.  
  38. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  39. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  40. 		ROI_flag = true;
  41. 	}
  42.  
  43. 	result = ThreadedRoundPipe(imgOriginal, top, bottom, canvas, debug);//调用识别算法
  44.  
  45. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  46. 	//img.copyTo(canvas);
  47.  
  48.  
  49. 	x值和y值补上ROI左上点x和y
  50. 	//if (ROI_flag) {
  51. 	//	if (Corner != cv::Point(0, 0))
  52. 	//	{
  53. 	//		Corner.x += ROI_tl.x;
  54. 	//		Corner.y += ROI_tl.y;
  55. 	//	}
  56. 	//}
  57.  
  58. 	img.copyTo(canvas);//调试时注释掉
  59.  
  60. 	cv::Mat t1;
  61. 	//canvas.convertTo(t1, -1, 900 / 100.0, 200 - 100);//第一次线性变换100
  62. 	canvas.convertTo(t1, -1, 900 / 100.0, 200 - 100);//第一次线性变换100
  63. 	if (debug)cv::imshow("线性变换ROI", t1);
  64. 	t1.copyTo(canvas);
  65.  
  66. 	if (result)
  67. 	{
  68. 		endRes = 1;
  69. 		top += ROI_tl;
  70. 		bottom += ROI_tl;
  71. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  72. 		int w = 38;
  73. 		int thickness = 4;
  74. 		cv::Point curPoint = top;
  75. 		cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(235, 135, 40), thickness);
  76. 		curPoint = bottom;
  77. 		cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(235, 135, 40), thickness);
  78. 		circle(canvas, top, 2, cv::Scalar(0, 69, 255), -1);
  79. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  80. 		circle(canvas, bottom, 2, cv::Scalar(0, 69, 255), -1);
  81. 		circle(canvas, bottom, 43, cv::Scalar(255, 169, 0), 1);
  82. 	}
  83. 	else
  84. 	{
  85. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  86. 	}
  87.  
  88.  
  89. 	cv::Mat ROIcanvas;
  90. 	img.copyTo(ROIcanvas);
  91. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  92. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  93. 	if (debug) circle(ROIcanvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  94. 	if (debug) circle(ROIcanvas, bottom, 63, cv::Scalar(0, 169, 255), 1);
  95. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  96. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  97. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  98. 	if (debug) imshow("ROI", ROIcanvas);
  99. 	if (debug) imshow("ROI_canvas", canvas);
  100. 	//ROIcanvas.copyTo(canvas);
  101.  
  102.  
  103. 	//cv::Rect roi_(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  104. 	//if(!ROI_flag) canvas.copyTo(canvas);
  105. 	//else canvas = canvas(roi_);	//裁剪出的ROI区域
  106.  
  107. 	return result;
  108. }

调用main：

  1. //螺纹圆管：ThreadedRoundPipe_ROI
  2. int main222() {
  3. 	bool flag = false;
  4. 	//std::string filePath = "E:\\vsproject\\勋仪交接\\标克艾芬达暖水管/0529img";
  5. 	//std::string filePath = "../img/ThreadedRoundPipe/0530img";
  6. 	std::string filePath = "../img/ThreadedRoundPipe/0606img";
  7.  
  8.  
  9. 	string resPath = "/output";
  10. 	//string resPath = filePath + "./output";
  11.  
  12.  
  13. 	cv::Point tlP = cv::Point(200, 340);
  14. 	cv::Point brP = cv::Point(950, 999);
  15. 	cv::Point top(90, 90);
  16. 	cv::Point bottom(90, 90);
  17. 	cv::Mat canvas;
  18. 	ImageProcessingTest m_ImageProcessing;
  19. 	ofstream ofs("log.txt", ios::out);
  20. 	if (ofs)
  21. 	{
  22. 		ofs.clear();
  23. 		ofs << "日志启动" << endl;
  24. 	}
  25.  
  26. 	vector<string> path_name;
  27.  
  28.  
  29. 	bool   lianxu = false; //循环测试
  30.  
  31. 	if (!lianxu)
  32. 	{
  33. 		//string path = filePath + cv::format("\\Image_20230606135229441.png");
  34. 		string path = filePath + cv::format("\\Image_20230606135308519.png");
  35. 		//string path = filePath + cv::format("\\Image_20230601143638388.png");
  36. 		//string path = filePath + cv::format("\\Image_20230601143505168.png");
  37. 		//string path = filePath + cv::format("\\Image_20230529095011455.png");
  38. 		cv::Mat src = cv::imread(path);
  39. 		src.copyTo(canvas);
  40. 		cv::namedWindow("main_OriginalIMG", cv::RECURS_FILTER);
  41. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  42. 		cv::imshow("main_OriginalIMG", src);
  43.  
  44. 		//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  45. 		cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  46.  
  47.  
  48. 		bool flag = m_ImageProcessing.ThreadedRoundPipe_ROI(src, ROI, top, bottom, canvas, true);
  49.  
  50. 		cout << "\nmain ==== flag " << flag << endl;
  51. 		if (flag) {
  52. 			cout << "top = " << top.x << " " << top.y << endl;
  53. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  54. 		}
  55. 		canvas.copyTo(src);
  56. 		//cout << "\nmain ==== i " << endl;
  57. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  58. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  59. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  60. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  61. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  62. 		cv::imshow("main-结果图src", src);
  63. 	}
  64. 	else
  65. 	{
  66. 		vector<string> path_name;
  67. 		getAllFiles(filePath, path_name);
  68. 		cout << "path_name.size()=" << path_name.size() << endl;
  69.  
  70. 		int number = 1;
  71. 		for (auto i : path_name)
  72. 		{
  73. 			cv::Mat src = cv::imread(filePath + "\\" + i);
  74. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  75. 			src.copyTo(canvas);
  76. 			cv::imshow("main_OriginalIMG", src);
  77.  
  78. 			//将结果图写入文件方便查看结果
  79. 			int len = filePath.length();
  80. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  81. 			cout << "i=" << i << endl;
  82. 			int time_start = clock();
  83. 			int time_end = clock();
  84.  
  85. 			//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  86. 			cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  87.  
  88.  
  89. 			bool flag = m_ImageProcessing.ThreadedRoundPipe_ROI(src, ROI, top, bottom, canvas, false);
  90.  
  91.  
  92. 			time_end = clock();
  93. 			printf("\n耗时 %d ms \n", time_end - time_start);
  94. 			cout << "\nmain ==== flag " << flag << endl;
  95. 			if (flag) {
  96. 				cout << "top = " << top.x << " " << top.y << endl;
  97. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  98. 			}
  99. 			canvas.copyTo(src);
  100. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  101. 			cv::imshow("main-结果图src", src);
  102. 			cv::waitKey(10);
  103.  
  104.  
  105. 			//将结果图写入文件方便查看结果
  106. 			char*  filePath_ = new char[len + 1];
  107. 			strcpy(filePath_, filePath.c_str());
  108.  
  109. 			len = filePath.length();
  110. 			char*  resPath_ = new char[len + 1];
  111. 			strcpy(resPath_, resPath.c_str());
  112. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  113.  
  114. 			string out = filePath + resPath;
  115. 			cout << "\nmain ==== i " << i << endl;
  116. 			out.append("\\").append(i);
  117.  
  118. 			cout << "number【" << number << "】res path=" << out << endl;
  119. 			cv::imwrite(out, src);
  120. 			number++;
  121.  
  122. 			cout << "path_name.size() = " << path_name.size() << endl;
  123. 			cout << "*********************************************************************************************************************" << endl;
  124. 			if (number == path_name.size()) break;
  125. 		}
  126. 	}
  127. 	cout << "main 循环结束！！！" << endl;
  128.  
  129. 	ofs.close();
  130. 	cv::waitKey(0);
  131. 	system("Pause");
  132. 	return 0;
  133. }

返回双点+1个点集

  1. 	/*
  2. 	//矩形位置  返回双点+骨干点集
  3. 	cv::Mat img, 原图
  4. 	cv::Rect ROI, 
  5. 	cv::Point& top, 输出焊点（图像上位于较上方的点）
  6. 	cv::Point& bottom, 输出焊点（图像上位于较下方的点）
  7. 	std::vector<cv::Point>& allPntList：直线部分非弧线的点集
  8. 	cv::Mat &canvas,效果图可视化
  9. 	*/
  10. 	bool rectangle2Pnt(cv::Mat imgOriginal, cv::Point& top, cv::Point& bottom, std::vector<cv::Point>& allPntList, cv::Mat &canvas, bool debug = false);
  11. 	bool rectangle2Pnt_ROI(cv::Mat img, cv::Rect ROI, cv::Point& top, cv::Point& bottom, std::vector<cv::Point>& allPntList, cv::Mat &canvas, bool debug = false);
  1. bool PTank_imgProcess::rectangle2Pnt_ROI(cv::Mat img, cv::Rect ROI, cv::Point & top, cv::Point & bottom, std::vector<cv::Point>& allPntList, cv::Mat & canvas, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7. 	bottom = cv::Point(0, 0);
  8. 	allPntList.clear();
  9.  
  10. 	cv::Mat imgOriginal;
  11. 	bool ROI_flag;//是否有给信息区
  12. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  13. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  14. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  15. 	{
  16. 		img.copyTo(imgOriginal);
  17. 		ROI_flag = false;
  18. 	}
  19. 	else
  20. 	{
  21. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  22. 		{
  23. 			printf("信息区两点疑似传反！");
  24. 			return false;
  25. 		}
  26. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  27. 		{
  28. 			printf("信息区的框选超出图像范围！");
  29. 			return false;
  30. 		}
  31. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  32. 		{
  33. 			printf("信息区大小超出图像大小！");
  34. 			return false;
  35. 		}
  36.  
  37. 		//此时不合法的ROI都已提前返回false
  38.  
  39. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  40. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  41. 		ROI_flag = true;
  42. 	}
  43.  
  44. 	result = rectangle2Pnt(imgOriginal, top, bottom, allPntList, canvas, debug);//调用识别算法
  45.  
  46.  
  47. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  48. 	//img.copyTo(canvas);
  49.  
  50. 	img.copyTo(canvas);//调试时注释掉
  51. 	if (result)
  52. 	{
  53. 		endRes = 1;
  54. 		top += ROI_tl;
  55. 		bottom += ROI_tl;
  56. 		for (int i = 0; i < allPntList.size(); i++) {
  57. 			allPntList.at(i) += ROI_tl;
  58. 		}
  59. 		int w = 5;
  60. 		for (int i = 0; i < allPntList.size(); i++) {
  61. 			cv::Point curPoint = allPntList.at(i);
  62. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(55, 195, 40), 1);
  63. 		}
  64. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  65. 		circle(canvas, top, 3, cv::Scalar(0, 69, 255), 1);
  66. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  67. 		circle(canvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  68. 		circle(canvas, bottom, 43, cv::Scalar(255, 169, 0), 1);
  69. 	}
  70. 	else
  71. 	{
  72. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  73. 	}
  74.  
  75.  
  76. 	cv::Mat ROIcanvas;
  77. 	img.copyTo(ROIcanvas);
  78. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  79. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  80. 	if (debug) circle(ROIcanvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  81. 	if (debug) circle(ROIcanvas, bottom, 63, cv::Scalar(0, 169, 255), 1);
  82. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  83. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  84. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  85. 	if (debug) imshow("ROI", ROIcanvas);
  86. 	if (debug) imshow("ROI_canvas", canvas);
  87. 	//ROIcanvas.copyTo(canvas);
  88.  
  89. 	return result;
  90. }

调用main：

  1. //矩形位置  返回双点+骨干点集
  2. int main/*Rectangle2Pnt_ROI*/() {
  3. 	bool flag;
  4. 	//std::string filePath = "../../../../Log08141641/TestImg/src";
  5. 	std::string filePath = "E://vsproject//压力罐\\img\\矩形";
  6.  
  7.  
  8. 	string resPath = "/output";
  9. 	//string resPath = filePath + "./output";
  10.  
  11.  
  12. 	cv::Point tlP = cv::Point(500, 292);
  13. 	cv::Point brP = cv::Point(750, 677);
  14. 	//cv::Rect ROI = cv::Rect(0, 0, 0, 0);
  15. 	//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  16. 	//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  17. 	//cv::Rect ROI = cv::Rect(539, 218, 113, 513);
  18. 	cv::Point top(90, 90);
  19. 	cv::Point bottom(90, 90);
  20. 	int outMidY;
  21. 	cv::Mat canvas;
  22. 	PTank_imgProcess m_ImageProcessing;
  23. 	ofstream ofs("log.txt", ios::out);
  24. 	if (ofs)
  25. 	{
  26. 		ofs.clear();
  27. 		ofs << "日志启动" << endl;
  28. 	}
  29. 	std::vector<cv::Point> allPntList;
  30. 	vector<string> path_name;
  31.  
  32.  
  33. 	bool   getROI = true;
  34. 	bool   lianxu = false; //循环测试
  35.  
  36. 	if (!lianxu)
  37. 	{
  38. 		string name = cv::format("\\Image_20240627144421072.png");
  39. 		string path = filePath + name;
  40. 		cv::Mat src = cv::imread(path);
  41. 		src.copyTo(canvas);
  42. 		cv::namedWindow("main_OriginalIMG", cv::RECURS_FILTER);
  43. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  44. 		cv::imshow("main_OriginalIMG", src);
  45.  
  46. 		cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  47. 		//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  48. 		if (getROI) ROI = getROIFromString(name);
  49.  
  50.  
  51. 		bool flag = m_ImageProcessing.rectangle2Pnt_ROI(src, ROI, top, bottom, allPntList, canvas, true);
  52.  
  53. 		cout << "\nmain ==== flag " << flag << endl;
  54. 		if (flag) {
  55. 			cout << "top = " << top.x << " " << top.y << endl;
  56. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  57. 		}
  58. 		canvas.copyTo(src);
  59. 		//cout << "\nmain ==== i " << endl;
  60. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  61. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  62. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  63. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  64. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  65. 		cv::imshow("main-结果图src", src);
  66. 		cv::namedWindow("main-canvas", cv::RECURS_FILTER);
  67. 		cvSetMouseCallback("main-canvas", on_mouse_singleClick, 0);
  68. 		cv::imshow("main-canvas", canvas);
  69. 	}
  70. 	else
  71. 	{
  72. 		vector<string> path_name;
  73. 		getAllFiles(filePath, path_name);
  74. 		cout << "path_name.size()=" << path_name.size() << endl;
  75.  
  76. 		int number = 1;
  77. 		for (auto i : path_name)
  78. 		{
  79. 			//cout << i.size() << endl;
  80. 			if (i.size() == 7) continue;
  81. 			if (i.find(".log") != std::string::npos) continue;
  82. 			std::string path11 = filePath + "\\" + i;
  83. 			cv::Mat src = cv::imread(path11);
  84. 			//cv::Mat src = cv::imread(filePath + "\\" + i);
  85. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  86. 			src.copyTo(canvas);
  87. 			cv::imshow("main_OriginalIMG", src);
  88.  
  89. 			//将结果图写入文件方便查看结果
  90. 			int len = filePath.length();
  91. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  92. 			cout << "i=" << i << ", path11=" << path11 << endl;
  93. 			int time_start = clock();
  94. 			int time_end = clock();
  95.  
  96. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  97. 			//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  98. 			if (getROI) ROI = getROIFromString(i);
  99.  
  100.  
  101. 			bool flag = m_ImageProcessing.rectangle2Pnt_ROI(src, ROI, top, bottom, allPntList, canvas, false);
  102.  
  103.  
  104. 			time_end = clock();
  105. 			printf("\n耗时 %d ms \n", time_end - time_start);
  106. 			cout << "\nmain ==== flag " << flag << endl;
  107. 			if (flag) {
  108. 				cout << "top = " << top.x << " " << top.y << endl;
  109. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  110. 			}
  111. 			canvas.copyTo(src);
  112. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  113. 			cv::imshow("main-结果图src", src);
  114. 			cv::imshow("main-canvas", canvas);
  115.  
  116.  
  117. 			//将结果图写入文件方便查看结果
  118. 			char*  filePath_ = new char[len + 1];
  119. 			strcpy(filePath_, filePath.c_str());
  120.  
  121. 			len = filePath.length();
  122. 			char*  resPath_ = new char[len + 1];
  123. 			strcpy(resPath_, resPath.c_str());
  124. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  125.  
  126. 			string out = filePath + resPath;
  127. 			cout << "\nmain ==== i " << i << endl;
  128. 			out.append("\\").append(i);
  129.  
  130. 			cout << "number【" << number << "】res path=" << out << endl;
  131. 			cv::imwrite(out, src);
  132. 			number++;
  133.  
  134. 			cout << "path_name.size() = " << path_name.size() << endl;
  135. 			cout << "*********************************************************************************************************************" << endl;
  136. 			cv::waitKey(10);
  137. 		}
  138. 		cout << "循环结束！！！ main " << endl;
  139. 	}
  140.  
  141. 	ofs.close();
  142. 	cv::waitKey(0);
  143. 	system("Pause");
  144. 	return 0;
  145. }

返回双点+三点集+左右/上下翻转再进入识别

注：该接口中 bottom, linearPntB无效！！！

根据传入参数判断是否左右翻转，判断识别的是上半部分还是下半部分，下半部分则需再做上下翻转再进入识别。

  1. 	/*
  2. 	//D型主管，方形支管 只处理上半部分或下半部分
  3. 	cv::Mat img, 原图
  4. 	cv::Rect ROI, 可以让右边的干扰直反光落进去，上下不用截，左右可适当截小一下减少识别时的计算时间
  5. 	std::vector<cv::Point>& linearPnt：直线部分非弧线的点集
  6. 	cv::Point& top, 输出焊点（图像上位于较上方的点）倒角弧形边边上的激光最边边点
  7. 	cv::Point& bottom, 输出焊点（图像上位于较下方的点）倒角弧形边边上的激光最边边点
  8. 	cv::Mat &canvas,效果图可视化
  9. 	bool arcTop_right = true;//圆弧顶点是否朝右的标志量，默认为true（true为朝右，false为朝左）
  10. 	bool isTopHalf = true;//识别区域是否为上半部分，上半部分为true，下半部分反之
  11. 	注：该接口中 bottom, linearPntB无效！！！
  12. 	*/
  13. 	bool OneSideSquare(cv::Mat imgOriginal, std::vector<cv::Point>& linearPntT, std::vector<cv::Point>& linearPntB, std::vector<cv::Point>& linearPntM, cv::Point& top, cv::Point& bottom, cv::Mat &canvas, bool debug = false);
  14. 	bool OneSideSquare_ROI(cv::Mat img, cv::Rect ROI, std::vector<cv::Point>& linearPntT, std::vector<cv::Point>& linearPntB, std::vector<cv::Point>& linearPntM, cv::Point& top, cv::Point& bottom, cv::Mat &canvas, bool arcTop_right = true, bool isTopHalf = true, bool debug = false);
  1. bool ImageProcessing::OneSideSquare_ROI(cv::Mat img, cv::Rect ROI, std::vector<cv::Point>& linearPntT, std::vector<cv::Point>& linearPntB, std::vector<cv::Point>& linearPntM, cv::Point & top, cv::Point & bottom, cv::Mat & canvas, bool arcTop_right, bool isTopHalf, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7. 	bottom = cv::Point(0, 0);
  8. 	linearPntT.clear();
  9. 	linearPntB.clear();
  10. 	linearPntM.clear();
  11.  
  12. 	cv::Mat imgOriginal;
  13. 	bool ROI_flag;//是否有给信息区
  14. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  15. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  16. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  17. 	{
  18. 		img.copyTo(imgOriginal);
  19. 		ROI_flag = false;
  20. 	}
  21. 	else
  22. 	{
  23. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  24. 		{
  25. 			printf("信息区两点疑似传反！");
  26. 			return false;
  27. 		}
  28. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  29. 		{
  30. 			printf("信息区的框选超出图像范围！");
  31. 			return false;
  32. 		}
  33. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  34. 		{
  35. 			printf("信息区大小超出图像大小！");
  36. 			return false;
  37. 		}
  38.  
  39. 		//此时不合法的ROI都已提前返回false
  40.  
  41. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  42. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  43. 		ROI_flag = true;
  44. 	}
  45.  
  46. 	if (arcTop_right) {//圆弧顶点朝右的情况
  47. 		if (isTopHalf) {
  48. 			result = OneSideSquare(imgOriginal, linearPntT, linearPntB, linearPntM, top, bottom, canvas, debug);//调用识别算法
  49. 		}
  50. 		else {
  51. 			cv::Mat fpMt;
  52. 			cv::flip(imgOriginal, fpMt, 0);		// 上下翻转 x对称
  53. 			result = OneSideSquare(fpMt, linearPntT, linearPntB, linearPntM, top, bottom, canvas, debug);//调用识别算法
  54. 			if (result) {
  55. 				top.y = fpMt.rows - top.y - 1;
  56. 				for (int i = 0; i < linearPntT.size(); i++) {
  57. 					linearPntT.at(i).y = fpMt.rows - linearPntT.at(i).y - 1;
  58. 				}
  59. 				for (int i = 0; i < linearPntM.size(); i++) {
  60. 					linearPntM.at(i).y = fpMt.rows - linearPntM.at(i).y - 1;
  61. 				}
  62. 			}
  63. 		}
  64. 	}
  65. 	else {//圆弧顶点朝左的情况
  66. 		cv::Mat flipMat;
  67. 		cv::flip(imgOriginal, flipMat, 1);			// 左右翻转 y对称
  68. 		if (debug) imshow("图像左右翻转", flipMat);
  69.  
  70. 		if (isTopHalf) {
  71. 			result = OneSideSquare(flipMat, linearPntT, linearPntB, linearPntM, top, bottom, canvas, debug);//调用识别算法
  72. 		}
  73. 		else {
  74. 			cv::Mat fpMt;
  75. 			cv::flip(flipMat, fpMt, 0);		// 上下翻转 x对称
  76. 			result = OneSideSquare(fpMt, linearPntT, linearPntB, linearPntM, top, bottom, canvas, debug);//调用识别算法
  77. 			if (result) {
  78. 				top.y = fpMt.rows - top.y - 1;
  79. 				for (int i = 0; i < linearPntT.size(); i++) {
  80. 					linearPntT.at(i).y = fpMt.rows - linearPntT.at(i).y - 1;
  81. 				}
  82. 				for (int i = 0; i < linearPntM.size(); i++) {
  83. 					linearPntM.at(i).y = fpMt.rows - linearPntM.at(i).y - 1;
  84. 				}
  85. 			}
  86. 		}
  87.  
  88. 		if (result) {
  89. 			top.x = flipMat.cols - top.x - 1;
  90. 			//bottom.x = flipMat.cols - bottom.x - 1;
  91. 			for (int i = 0; i < linearPntT.size(); i++) {
  92. 				linearPntT.at(i).x = flipMat.cols - linearPntT.at(i).x - 1;
  93. 			}
  94. 			for (int i = 0; i < linearPntB.size(); i++) {
  95. 				linearPntB.at(i).x = flipMat.cols - linearPntB.at(i).x - 1;
  96. 			}
  97. 			for (int i = 0; i < linearPntM.size(); i++) {
  98. 				linearPntM.at(i).x = flipMat.cols - linearPntM.at(i).x - 1;
  99. 			}
  100. 		}
  101. 	}
  102.  
  103.  
  104. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  105. 	//img.copyTo(canvas);
  106.  
  107.  
  108. 	x值和y值补上ROI左上点x和y
  109. 	//if (ROI_flag) {
  110. 	//	if (Corner != cv::Point(0, 0))
  111. 	//	{
  112. 	//		Corner.x += ROI_tl.x;
  113. 	//		Corner.y += ROI_tl.y;
  114. 	//	}
  115. 	//}
  116.  
  117. 	img.copyTo(canvas);//调试时注释掉
  118. 	if (result)
  119. 	{
  120. 		endRes = 1;
  121. 		top += ROI_tl;
  122. 		//bottom += ROI_tl;
  123. 		for (int i = 0; i < linearPntT.size(); i++) {
  124. 			linearPntT.at(i) += ROI_tl;
  125. 		}
  126. 		for (int i = 0; i < linearPntB.size(); i++) {
  127. 			linearPntB.at(i) += ROI_tl;
  128. 		}
  129. 		for (int i = 0; i < linearPntM.size(); i++) {
  130. 			linearPntM.at(i) += ROI_tl;
  131. 		}
  132. 		for (int i = 0; i < linearPntT.size(); i++) {
  133. 			cv::Point curPoint = linearPntT.at(i);
  134. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - 5, curPoint.y), cv::Scalar(55, 195, 40), 1);
  135. 		}
  136. 		for (int i = 0; i < linearPntB.size(); i++) {
  137. 			cv::Point curPoint = linearPntB.at(i);
  138. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - 5, curPoint.y), cv::Scalar(55, 195, 40), 1);
  139. 		}
  140. 		for (int i = 0; i < linearPntM.size(); i++) {
  141. 			cv::Point curPoint = linearPntM.at(i);
  142. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - 5, curPoint.y), cv::Scalar(145, 175, 40), 1);
  143. 		}
  144. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  145. 		circle(canvas, top, 3, cv::Scalar(0, 69, 255), 1);
  146. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  147. 		//circle(canvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  148. 		//circle(canvas, bottom, 43, cv::Scalar(255, 169, 0), 1);
  149. 	}
  150. 	else
  151. 	{
  152. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  153. 	}
  154.  
  155.  
  156. 	cv::Mat ROIcanvas;
  157. 	img.copyTo(ROIcanvas);
  158. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  159. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  160. 	//if (debug) circle(ROIcanvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  161. 	//if (debug) circle(ROIcanvas, bottom, 63, cv::Scalar(0, 169, 255), 1);
  162. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  163. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  164. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  165. 	if (debug) imshow("ROI", ROIcanvas);
  166. 	if (debug) imshow("ROI_canvas", canvas);
  167. 	//ROIcanvas.copyTo(canvas);
  168.  
  169. 	return result;
  170. }

调用main：

  1. //D型主管，方形支管 上或下单侧  OneSideSquare_ROI
  2. int main() {
  3. 	bool flag;
  4. 	//std::string filePath = "../../../../0方管支管/1023img";
  5. 	std::string filePath = "../../../../0方管支管/left";
  6. 	//std::string filePath = "../../../../0方管支管/right";
  7.  
  8.  
  9. 	string resPath = "/output";
  10. 	//string resPath = filePath + "./output";
  11.  
  12.  
  13. 	//cv::Point tlP = cv::Point(500, 40);
  14. 	//cv::Point brP = cv::Point(750, 777);
  15. 	cv::Point tlP = cv::Point(584, 236);
  16. 	cv::Point brP = cv::Point(660, 820);
  17. 	//cv::Rect ROI = cv::Rect(0, 0, 0, 0);
  18. 	//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  19. 	//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  20. 	//cv::Rect ROI = cv::Rect(639, 188, 125, 569);
  21. 	cv::Point top(90, 90);
  22. 	cv::Point bottom(90, 90);
  23. 	cv::Point2f midP(90, 90);
  24. 	int outMidY;
  25. 	cv::Mat canvas;
  26. 	ImageProcessing m_ImageProcessing;
  27. 	ofstream ofs("log.txt", ios::out);
  28. 	if (ofs)
  29. 	{
  30. 		ofs.clear();
  31. 		ofs << "日志启动" << endl;
  32. 	}
  33. 	std::vector<cv::Point> linearPntT;
  34. 	std::vector<cv::Point> linearPntB;
  35. 	std::vector<cv::Point> linearPntM;
  36. 	vector<Picture_set> img_buf;
  37. 	vector<string> path_name;
  38. 	bool arcTop_right = !true;
  39. 	bool isTopHalf = true;
  40. 	int midyTolerant = 4;
  41.  
  42. 	bool   getROI = true;
  43. 	bool   lianxu = false; //循环测试
  44.  
  45. 	if (lianxu)
  46. 	{
  47. 		//string path = filePath + cv::format("\\Image_20231023105829348.png");
  48. 		//std::string name = cv::format("\\ImgD_2023_10_26_10_48_12_135_567_196_153_603_Src.png");
  49. 		std::string name = cv::format("\\ImgD_2023_10_25_18_16_24_519_569_200_166_559_Src.png");
  50. 		string path = filePath + name;
  51. 		cv::Mat src = cv::imread(path);
  52. 		src.copyTo(canvas);
  53. 		cv::namedWindow("main_OriginalIMG", cv::NORMCONV_FILTER);
  54. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  55. 		cv::imshow("main_OriginalIMG", src);
  56.  
  57. 		//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  58. 		cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  59. 		if (getROI) {
  60. 			ROI = getROIFromString(name);
  61. 			ROI.height /= 2;
  62. 			if (!isTopHalf) {
  63. 				ROI.y += ROI.height;
  64. 			}
  65. 		}
  66.  
  67. 		//鼠标点击的坐标【691, 319】
  68. 		//	鼠标点击的坐标【692, 632】
  69. 		//cout << src.size() << endl;
  70. 		//rotateImage(src);
  71. 		//cout << src.size() << endl;
  72. 		//鼠标点击的坐标【707, 691】
  73. 		//	鼠标点击的坐标【395, 691】
  74. 		//trueX=outY;
  75. 		//trueY=oriSrc.rows-outX;
  76. 		//circle(src, cv::Point(622, 424), 5, cv::Scalar(255, 255, 255), -1);
  77. 		//cv::Mat flipMat;
  78. 		//cv::flip(src, flipMat, 1);
  79. 		//cv::Point q = cv::Point(flipMat.cols - 622 - 1, 424)/* - cv::Point(546, 228)*/;
  80. 		//circle(flipMat, q, 3, cv::Scalar(0, 0, 0), -1);
  81. 		//cv::imshow("flipMat", flipMat);
  82. 		//cv::waitKey();
  83. 		//cv::flip(src, src, 0);
  84.  
  85. 		bool flag = m_ImageProcessing.OneSideSquare_ROI(src, ROI, linearPntT, linearPntB, linearPntM, top, bottom, canvas, arcTop_right, isTopHalf, true);
  86.  
  87. 		cout << "\nmain ==== flag " << flag << endl;
  88. 		if (flag) {
  89. 			cout << "top = " << top.x << " " << top.y << endl;
  90. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  91. 		}
  92. 		canvas.copyTo(src);
  93. 		//cout << "\nmain ==== i " << endl;
  94. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  95. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  96. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  97. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  98. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  99. 		cv::imshow("main-结果图src", src);
  100. 		cv::namedWindow("main-canvas", cv::RECURS_FILTER);
  101. 		cvSetMouseCallback("main-canvas", on_mouse_singleClick, 0);
  102. 		cv::imshow("main-canvas", canvas);
  103. 	}
  104. 	else
  105. 	{
  106. 		vector<string> path_name;
  107. 		getAllFiles(filePath, path_name);
  108. 		cout << "path_name.size()=" << path_name.size() << endl;
  109.  
  110. 		int number = 1;
  111. 		for (auto i : path_name)
  112. 		{
  113. 			//cout << i.size() << endl;
  114. 			if (i.size() == 7) continue;
  115. 			if (i.find(".log") != std::string::npos) continue;
  116. 			if (i.find(".txt") != std::string::npos) continue;
  117. 			std::string name = i;
  118. 			std::string path11 = filePath + "\\" + i;
  119. 			cv::Mat src = cv::imread(path11);
  120. 			//cv::Mat src = cv::imread(filePath + "\\" + i);
  121. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  122. 			src.copyTo(canvas);
  123. 			cv::imshow("main_OriginalIMG", src);
  124.  
  125. 			//将结果图写入文件方便查看结果
  126. 			int len = filePath.length();
  127. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  128. 			cout << "i=" << i << ", path11=" << path11 << endl;
  129. 			int time_start = clock();
  130. 			int time_end = clock();
  131.  
  132. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  133. 			cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  134. 			if (getROI) {
  135. 				ROI = getROIFromString(name);
  136. 				ROI.height /= 2;
  137. 				if (!isTopHalf) {
  138. 					ROI.y += ROI.height;
  139. 				}
  140. 			}
  141.  
  142. 			bool flag = m_ImageProcessing.OneSideSquare_ROI(src, ROI, linearPntT, linearPntB, linearPntM, top, bottom, canvas, arcTop_right, isTopHalf, false);
  143.  
  144.  
  145. 			time_end = clock();
  146. 			printf("\n耗时 %d ms \n", time_end - time_start);
  147. 			cout << "\nmain ==== flag " << flag << endl;
  148. 			if (flag) {
  149. 				cout << "top = " << top.x << " " << top.y << endl;
  150. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  151. 			}
  152. 			canvas.copyTo(src);
  153. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  154. 			cv::imshow("main-结果图src", src);
  155. 			cv::imshow("main-canvas", canvas);
  156.  
  157.  
  158. 			//将结果图写入文件方便查看结果
  159. 			char*  filePath_ = new char[len + 1];
  160. 			strcpy(filePath_, filePath.c_str());
  161.  
  162. 			len = filePath.length();
  163. 			char*  resPath_ = new char[len + 1];
  164. 			strcpy(resPath_, resPath.c_str());
  165. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  166.  
  167. 			string out = filePath + resPath;
  168. 			cout << "\nmain ==== i " << i << endl;
  169. 			out.append("\\").append(i);
  170.  
  171. 			cout << "number【" << number << "】res path=" << out << endl;
  172. 			cv::imwrite(out, src);
  173. 			number++;
  174.  
  175. 			cout << "path_name.size() = " << path_name.size() << endl;
  176. 			cout << "*********************************************************************************************************************" << endl;
  177. 			cv::waitKey(10);
  178. 		}
  179. 		cout << "循环结束！！！ main " << endl;
  180. 	}
  181.  
  182. 	ofs.close();
  183. 	cv::waitKey(0);
  184. 	system("Pause");
  185. 	return 0;
  186. }

三点

返回三点+1个点集  

  1. 	/*
  2. 	//两弧线夹圆弧  返回三点+骨干点集
  3. 	cv::Mat img, 原图
  4. 	cv::Rect ROI,
  5. 	cv::Point& top, 输出焊点（图像上位于较上方的点）
  6. 	cv::Point& bottom, 输出焊点（图像上位于较下方的点）
  7. 	cv::Point2f& midP, 圆弧中点
  8. 	std::vector<cv::Point>& allPntList：直线部分非弧线的点集
  9. 	cv::Mat &canvas,效果图可视化
  10. 	*/
  11. 	bool twoArcsClipTheArc(cv::Mat imgOriginal, cv::Point& top, cv::Point& bottom, cv::Point2f& midP, std::vector<cv::Point>& allPntList, cv::Mat &canvas, bool debug = false);
  12. 	bool twoArcsClipTheArc_ROI(cv::Mat img, cv::Rect ROI, cv::Point& top, cv::Point& bottom, cv::Point2f& midP, std::vector<cv::Point>& allPntList, cv::Mat &canvas, bool arcTop_right = true, bool debug = false);
  1. bool PTank_imgProcess::twoArcsClipTheArc_ROI(cv::Mat img, cv::Rect ROI, cv::Point & top, cv::Point & bottom, cv::Point2f & midP, std::vector<cv::Point>& allPntList, cv::Mat & canvas, bool arcTop_right, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7. 	bottom = cv::Point(0, 0);
  8. 	midP = cv::Point2f(0, 0);
  9. 	allPntList.clear();
  10.  
  11. 	cv::Mat imgOriginal;
  12. 	bool ROI_flag;//是否有给信息区
  13. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  14. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  15. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  16. 	{
  17. 		img.copyTo(imgOriginal);
  18. 		ROI_flag = false;
  19. 	}
  20. 	else
  21. 	{
  22. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  23. 		{
  24. 			printf("信息区两点疑似传反！");
  25. 			return false;
  26. 		}
  27. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  28. 		{
  29. 			printf("信息区的框选超出图像范围！");
  30. 			return false;
  31. 		}
  32. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  33. 		{
  34. 			printf("信息区大小超出图像大小！");
  35. 			return false;
  36. 		}
  37.  
  38. 		//此时不合法的ROI都已提前返回false
  39.  
  40. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  41. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  42. 		ROI_flag = true;
  43. 	}
  44.  
  45. 	if (arcTop_right) {//圆弧顶点朝右的情况
  46. 		result = twoArcsClipTheArc(imgOriginal, top, bottom, midP, allPntList, canvas, debug);//调用识别算法
  47. 	}
  48. 	else {//圆弧顶点朝左的情况
  49. 		cv::Mat flipMat;
  50. 		cv::flip(imgOriginal, flipMat, 1);			// 左右翻转 y对称
  51. 		if (debug) imshow("图像左右翻转", flipMat);
  52. 		result = twoArcsClipTheArc(flipMat, top, bottom, midP, allPntList, canvas, debug);//调用识别算法
  53. 		if (result) {
  54. 			top.x = flipMat.cols - top.x - 1;
  55. 			bottom.x = flipMat.cols - bottom.x - 1;
  56. 			for (int i = 0; i < allPntList.size(); i++) {
  57. 				allPntList.at(i).x = flipMat.cols - allPntList.at(i).x - 1;
  58. 			}
  59. 			if (midP != cv::Point2f(0, 0)) midP.x = flipMat.cols - midP.x - 1;
  60. 		}
  61. 	}
  62.  
  63.  
  64. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  65. 	if (debug) printf("                                                roiF  midP=【%f, %f】\n", midP.x, midP.y);
  66. 	//img.copyTo(canvas);
  67.  
  68. 	//img.copyTo(canvas);//调试时注释掉
  69. 	if (result)
  70. 	{
  71. 		endRes = 1;
  72. 		for (int i = 0; i < allPntList.size(); i++) {
  73. 			allPntList.at(i) += ROI_tl;
  74. 		}
  75. 		for (int i = 0; i < allPntList.size(); i++) {
  76. 			cv::Point curPoint = allPntList.at(i);
  77. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - 5, curPoint.y), cv::Scalar(55, 195, 40), 1);
  78. 		}
  79. 		top += ROI_tl;
  80. 		bottom += ROI_tl;
  81. 		midP += cv::Point2f(ROI_tl);
  82. 		//if (midP != cv::Point2f(0, 0)) {
  83. 		//	midP.x = midP.x + ROI_tl.x;
  84. 		//	midP.y = midP.y + ROI_tl.y;
  85. 		//}
  86. 		circle(canvas, midP, 3, cv::Scalar(0, 69, 255), 1);
  87. 		circle(canvas, midP, 43, cv::Scalar(0, 169, 255), 1);
  88. 		cv::line(canvas, midP, cv::Point2f(5, midP.y), cv::Scalar(155, 195, 40), 1);
  89. 		cv::line(canvas, cv::Point2f(midP.x, midP.y - 5), cv::Point2f(midP.x, midP.y + 5), cv::Scalar(155, 195, 40), 1);
  90. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  91. 		circle(canvas, top, 3, cv::Scalar(0, 69, 255), 1);
  92. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  93. 		circle(canvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  94. 		circle(canvas, bottom, 43, cv::Scalar(255, 169, 0), 1);
  95. 	}
  96. 	else
  97. 	{
  98. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  99. 	}
  100.  
  101.  
  102. 	cv::Mat ROIcanvas;
  103. 	img.copyTo(ROIcanvas);
  104. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  105. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  106. 	if (debug) circle(ROIcanvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  107. 	if (debug) circle(ROIcanvas, bottom, 63, cv::Scalar(0, 169, 255), 1);
  108. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  109. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  110. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  111. 	if (debug) printf("                                                roiF  midP=【%f, %f】\n", midP.x, midP.y);
  112. 	if (debug) imshow("ROI", ROIcanvas);
  113. 	if (debug) imshow("ROI_canvas", canvas);
  114. 	//ROIcanvas.copyTo(canvas);
  115.  
  116. 	return result;
  117. }

 调用main：

  1. //两弧线夹圆弧  返回三点+骨干点集
  2. int main/*TwoArcsClipTheArc*/() {
  3. 	bool flag;
  4. 	std::string filePath = "../../img/圆弧夹圆弧";
  5.  
  6. 	string resPath = "/output";
  7. 	//string resPath = filePath + "./output";
  8.  
  9.  
  10. 	cv::Point tlP = cv::Point(500, 292);
  11. 	cv::Point brP = cv::Point(750, 677);
  12. 	//cv::Rect ROI = cv::Rect(0, 0, 0, 0);
  13. 	//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  14. 	//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  15. 	//cv::Rect ROI = cv::Rect(539, 218, 113, 513);
  16. 	cv::Rect ROI = cv::Rect(546, 228, 101, 520);
  17. 	cv::Point top(90, 90);
  18. 	cv::Point bottom(90, 90);
  19. 	cv::Point2f midP(90, 90);
  20. 	int outMidY;
  21. 	cv::Mat canvas;
  22. 	PTank_imgProcess m_ImageProcessing;
  23. 	ofstream ofs("log.txt", ios::out);
  24. 	if (ofs)
  25. 	{
  26. 		ofs.clear();
  27. 		ofs << "日志启动" << endl;
  28. 	}
  29. 	std::vector<cv::Point> allPntList;
  30. 	vector<string> path_name;
  31. 	bool arcTop_right = true;
  32.  
  33. 	bool   getROI = true;
  34. 	bool   lianxu = false; //循环测试
  35.  
  36. 	if (!lianxu)
  37. 	{
  38. 		string name = cv::format("\\Image_20240702114054217.png");
  39. 		//string name = cv::format("\\Image_20240702170558613.png");//斜
  40. 		//string name = cv::format("\\Image_20240702170644641.png");//斜
  41. 		string path = filePath + name;
  42. 		//string path = filePath + cv::format("\\ImgD_2023_08_17_15_06_45_412_639_188_125_569_Src.png");
  43. 		cv::Mat src = cv::imread(path);
  44. 		src.copyTo(canvas);
  45. 		cv::namedWindow("main_OriginalIMG", cv::NORMCONV_FILTER);
  46. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  47. 		cv::imshow("main_OriginalIMG", src);
  48.  
  49. 		//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  50. 		//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  51. 		if (getROI) ROI = getROIFromString(name);
  52.  
  53.  
  54. 		bool flag = m_ImageProcessing.twoArcsClipTheArc_ROI(src, ROI, top, bottom, midP, allPntList, canvas, arcTop_right, true);
  55.  
  56. 		cout << "\nmain ==== flag " << flag << endl;
  57. 		if (flag) {
  58. 			cout << "top = " << top.x << " " << top.y << endl;
  59. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  60. 			cout << "midP = " << midP.x << " " << midP.y << endl;
  61. 		}
  62. 		canvas.copyTo(src);
  63. 		//cout << "\nmain ==== i " << endl;
  64. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  65. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  66. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  67. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  68. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  69. 		cv::imshow("main-结果图src", src);
  70. 		cv::namedWindow("main-canvas", cv::RECURS_FILTER);
  71. 		cvSetMouseCallback("main-canvas", on_mouse_singleClick, 0);
  72. 		cv::imshow("main-canvas", canvas);
  73. 	}
  74. 	else
  75. 	{
  76. 		vector<string> path_name;
  77. 		getAllFiles(filePath, path_name);
  78. 		cout << "path_name.size()=" << path_name.size() << endl;
  79.  
  80. 		int number = 1;
  81. 		for (auto i : path_name)
  82. 		{
  83. 			//cout << i.size() << endl;
  84. 			if (i.size() == 7) continue;
  85. 			if (i.find(".log") != std::string::npos) continue;
  86. 			if (i.find(".txt") != std::string::npos) continue;
  87. 			if (i.find("_result") != std::string::npos) continue;
  88. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  89. 			std::string path11 = filePath + "\\" + i;
  90. 			cv::Mat src = cv::imread(path11);
  91. 			//cv::Mat src = cv::imread(filePath + "\\" + i);
  92. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  93. 			src.copyTo(canvas);
  94. 			cv::imshow("main_OriginalIMG", src);
  95.  
  96. 			//将结果图写入文件方便查看结果
  97. 			int len = filePath.length();
  98. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  99. 			cout << "i=" << i << ", path11=" << path11 << endl;
  100. 			int time_start = clock();
  101. 			int time_end = clock();
  102.  
  103. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  104. 			//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  105. 			if (getROI) ROI = getROIFromString(i);
  106.  
  107.  
  108. 			bool flag = m_ImageProcessing.twoArcsClipTheArc_ROI(src, ROI, top, bottom, midP, allPntList, canvas, arcTop_right, false);
  109.  
  110.  
  111. 			time_end = clock();
  112. 			printf("\n耗时 %d ms \n", time_end - time_start);
  113. 			cout << "\nmain ==== flag " << flag << endl;
  114. 			if (flag) {
  115. 				cout << "top = " << top.x << " " << top.y << endl;
  116. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  117. 			}
  118. 			canvas.copyTo(src);
  119. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  120. 			cv::imshow("main-结果图src", src);
  121. 			cv::imshow("main-canvas", canvas);
  122.  
  123.  
  124. 			//将结果图写入文件方便查看结果
  125. 			char*  filePath_ = new char[len + 1];
  126. 			strcpy(filePath_, filePath.c_str());
  127.  
  128. 			len = filePath.length();
  129. 			char*  resPath_ = new char[len + 1];
  130. 			strcpy(resPath_, resPath.c_str());
  131. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  132.  
  133. 			string out = filePath + resPath;
  134. 			cout << "\nmain ==== i " << i << endl;
  135. 			out.append("\\").append(i);
  136.  
  137. 			cout << "number【" << number << "】res path=" << out << endl;
  138. 			cv::imwrite(out, src);
  139. 			number++;
  140.  
  141. 			cout << "path_name.size() = " << path_name.size() << endl;
  142. 			cout << "*********************************************************************************************************************" << endl;
  143. 			cv::waitKey(10);
  144. 		}
  145. 		cout << "循环结束！！！ main " << endl;
  146. 	}
  147.  
  148. 	ofs.close();
  149. 	cv::waitKey(0);
  150. 	system("Pause");
  151. 	return 0;
  152. }

返回三点+双点集+左右翻转再进入识别

左右翻转变回来后必须要x-1，不然会与原来基于原图上的坐标对应不上。

  1. 	/*
  2.     //D型管的两直线夹圆弧
  3. 	//周六新策略：两边直线中间圆弧，激光平行于主管 (__O__两端直线中间半圆弧找圆弧的两端拐点)---圆轨迹
  4. 	cv::Mat img, 原图
  5. 	cv::Rect ROI, 可以让右边的干扰直反光落进去，上下不用截，左右可适当截小一下减少识别时的计算时间
  6. 	int outMidY,因y偏差过大时返回的两点无效，则返回圆弧区域的中心y
  7. 	cv::Point& midP, outMidY对应的圆弧上的骨干点（midP为(0,0)时，该值无效）
  8. 	std::vector<cv::Point>& linearPnt：直线部分非弧线的点集
  9. 	cv::Point& top, 输出焊点（图像上位于较上方的点）
  10. 	cv::Point& bottom, 输出焊点（图像上位于较下方的点）
  11. 	cv::Mat &canvas,效果图可视化
  12. 	int midyTolerant = 2;//outMidY的容差范围
  13. 	bool arcTop_right = true;//圆弧顶点是否朝右的标志量，默认为true（true为朝右，false为朝左）
  14. 	*/
  15. 	bool LaserParallelToTheMainD(cv::Mat imgOriginal, int &outMidY, cv::Point2f& midP, std::vector<cv::Point>& linearPntT, std::vector<cv::Point>& linearPntB, cv::Point& top, cv::Point& bottom, cv::Mat &canvas, int midyTolerant, bool debug = false);
  16. 	bool LaserParallelToTheMainD_ROI(cv::Mat img, cv::Rect ROI, int &outMidY, cv::Point2f& midP, std::vector<cv::Point>& linearPntT, std::vector<cv::Point>& linearPntB, cv::Point& top, cv::Point& bottom, cv::Mat &canvas, int midyTolerant = 2, bool arcTop_right = true, bool debug = false);
  1. bool ImageProcessing::LaserParallelToTheMainD_ROI(cv::Mat img, cv::Rect ROI, int &outMidY, cv::Point2f& midP, std::vector<cv::Point>& linearPntT, std::vector<cv::Point>& linearPntB, cv::Point & top, cv::Point & bottom, cv::Mat & canvas, int midyTolerant, bool arcTop_right, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7. 	bottom = cv::Point(0, 0);
  8. 	midP = cv::Point2f(0, 0);
  9. 	linearPntT.clear();
  10. 	linearPntB.clear();
  11. 	outMidY = 0;
  12.  
  13. 	cv::Mat imgOriginal;
  14. 	bool ROI_flag;//是否有给信息区
  15. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  16. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  17. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  18. 	{
  19. 		img.copyTo(imgOriginal);
  20. 		ROI_flag = false;
  21. 	}
  22. 	else
  23. 	{
  24. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  25. 		{
  26. 			printf("信息区两点疑似传反！");
  27. 			return false;
  28. 		}
  29. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  30. 		{
  31. 			printf("信息区的框选超出图像范围！");
  32. 			return false;
  33. 		}
  34. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  35. 		{
  36. 			printf("信息区大小超出图像大小！");
  37. 			return false;
  38. 		}
  39.  
  40. 		//此时不合法的ROI都已提前返回false
  41.  
  42. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  43. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  44. 		ROI_flag = true;
  45. 	}
  46.  
  47. 	if (arcTop_right) {//圆弧顶点朝右的情况
  48. 		result = LaserParallelToTheMainD(imgOriginal, outMidY, midP, linearPntT, linearPntB, top, bottom, canvas, midyTolerant, debug);//调用识别算法
  49. 	}
  50. 	else {//圆弧顶点朝左的情况
  51. 		cv::Mat flipMat;
  52. 		cv::flip(imgOriginal, flipMat, 1);			// 左右翻转 y对称
  53. 		if (debug) imshow("图像左右翻转", flipMat);
  54. 		result = LaserParallelToTheMainD(flipMat, outMidY, midP, linearPntT, linearPntB, top, bottom, canvas, midyTolerant, debug);//调用识别算法
  55. 		if (result) {
  56. 			top.x = flipMat.cols - top.x - 1;
  57. 			bottom.x = flipMat.cols - bottom.x - 1;
  58. 			for (int i = 0; i < linearPntT.size(); i++) {
  59. 				linearPntT.at(i).x = flipMat.cols - linearPntT.at(i).x - 1;
  60. 			}
  61. 			for (int i = 0; i < linearPntB.size(); i++) {
  62. 				linearPntB.at(i).x = flipMat.cols - linearPntB.at(i).x - 1;
  63. 			}
  64. 			if (midP != cv::Point2f(0, 0)) midP.x = flipMat.cols - midP.x - 1;
  65. 		}
  66. 	}
  67.  
  68.  
  69. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  70. 	if (debug) printf("                                                roiF  midP=【%f, %f】\n", midP.x, midP.y);
  71. 	//img.copyTo(canvas);
  72.  
  73.  
  74. 	x值和y值补上ROI左上点x和y
  75. 	//if (ROI_flag) {
  76. 	//	if (Corner != cv::Point(0, 0))
  77. 	//	{
  78. 	//		Corner.x += ROI_tl.x;
  79. 	//		Corner.y += ROI_tl.y;
  80. 	//	}
  81. 	//}
  82.  
  83. 	img.copyTo(canvas);//调试时注释掉
  84. 	if (result)
  85. 	{
  86. 		endRes = 1;
  87. 		top += ROI_tl;
  88. 		bottom += ROI_tl;
  89. 		if (midP != cv::Point2f(0, 0)) {
  90. 			midP.x = midP.x + ROI_tl.x;
  91. 			midP.y = midP.y + ROI_tl.y;
  92. 		}
  93. 		cv::line(canvas, midP, cv::Point2f(5, midP.y), cv::Scalar(55, 195, 40), 1);
  94. 		cv::line(canvas, cv::Point2f(midP.x, midP.y - 5), cv::Point2f(midP.x, midP.y + 5), cv::Scalar(55, 195, 40), 1);
  95. 		for (int i = 0; i < linearPntT.size(); i++) {
  96. 			linearPntT.at(i) += ROI_tl;
  97. 		}
  98. 		for (int i = 0; i < linearPntB.size(); i++) {
  99. 			linearPntB.at(i) += ROI_tl;
  100. 		}
  101. 		for (int i = 0; i < linearPntT.size(); i++) {
  102. 			cv::Point curPoint = linearPntT.at(i);
  103. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - 5, curPoint.y), cv::Scalar(55, 195, 40), 1);
  104. 		}
  105. 		for (int i = 0; i < linearPntB.size(); i++) {
  106. 			cv::Point curPoint = linearPntB.at(i);
  107. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - 5, curPoint.y), cv::Scalar(55, 195, 40), 1);
  108. 		}
  109. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  110. 		circle(canvas, top, 3, cv::Scalar(0, 69, 255), 1);
  111. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  112. 		circle(canvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  113. 		circle(canvas, bottom, 43, cv::Scalar(255, 169, 0), 1);
  114. 	}
  115. 	else
  116. 	{
  117. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  118. 	}
  119.  
  120.  
  121. 	cv::Mat ROIcanvas;
  122. 	img.copyTo(ROIcanvas);
  123. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  124. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  125. 	if (debug) circle(ROIcanvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  126. 	if (debug) circle(ROIcanvas, bottom, 63, cv::Scalar(0, 169, 255), 1);
  127. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  128. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  129. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  130. 	if (debug) printf("                                                roiF  midP=【%f, %f】\n", midP.x, midP.y);
  131. 	if (debug) imshow("ROI", ROIcanvas);
  132. 	if (debug) imshow("ROI_canvas", canvas);
  133. 	//ROIcanvas.copyTo(canvas);
  134.  
  135. 	return result;
  136. }

调用main：

  1. //CXL 两边直线中间圆弧，激光平行于主管 DDDDDDDDDDDDDDDDD 圆轨迹  增加直线部分点集 的重载
  2. int main() {
  3. 	bool flag;
  4. 	std::string filePath = "../../../../Log08141641/TestImg/src";
  5.  
  6.  
  7. 	string resPath = "/output";
  8. 	//string resPath = filePath + "./output";
  9.  
  10.  
  11. 	cv::Point tlP = cv::Point(500, 292);
  12. 	cv::Point brP = cv::Point(750, 677);
  13. 	//cv::Rect ROI = cv::Rect(0, 0, 0, 0);
  14. 	//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  15. 	//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  16. 	//cv::Rect ROI = cv::Rect(539, 218, 113, 513);
  17. 	cv::Point top(90, 90);
  18. 	cv::Point bottom(90, 90);
  19. 	cv::Point2f midP(90, 90);
  20. 	int outMidY;
  21. 	cv::Mat canvas;
  22. 	ImageProcessing m_ImageProcessing;
  23. 	ofstream ofs("log.txt", ios::out);
  24. 	if (ofs)
  25. 	{
  26. 		ofs.clear();
  27. 		ofs << "日志启动" << endl;
  28. 	}
  29. 	std::vector<cv::Point> linearPntT;
  30. 	std::vector<cv::Point> linearPntB;
  31. 	vector<string> path_name;
  32. 	bool arcTop_right = !true;
  33. 	int midyTolerant = 4;
  34.  
  35. 	bool   getROI = true;
  36. 	bool   lianxu = false; //循环测试
  37.  
  38. 	if (!lianxu)
  39. 	{
  40. 		string name = cv::format("\\ImgD_2024_06_04_09_14_48_111_565_82_86_851_Src.png");
  41. 		string path = filePath + name;
  42. 		//string path = filePath + cv::format("\\ImgD_2023_08_17_15_06_45_412_639_188_125_569_Src.png");
  43. 		cv::Mat src = cv::imread(path);
  44. 		src.copyTo(canvas);
  45. 		cv::namedWindow("main_OriginalIMG", cv::NORMCONV_FILTER);
  46. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  47. 		cv::imshow("main_OriginalIMG", src);
  48.  
  49. 		//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  50. 		//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  51. 		if (getROI) ROI = getROIFromString(name);
  52.  
  53. 		//trueX=outY;
  54. 		//trueY=oriSrc.rows-outX;
  55. 		//circle(src, cv::Point(622, 424), 5, cv::Scalar(255, 255, 255), -1);
  56. 		//cv::Mat flipMat;
  57. 		//cv::flip(src, flipMat, 1);
  58. 		//cv::Point q = cv::Point(flipMat.cols - 622 - 1, 424)/* - cv::Point(546, 228)*/;
  59. 		//circle(flipMat, q, 3, cv::Scalar(0, 0, 0), -1);
  60. 		//cv::imshow("flipMat", flipMat);
  61. 		//cv::waitKey();
  62.  
  63. 		bool flag = m_ImageProcessing.LaserParallelToTheMainD_ROI(src, ROI, outMidY, midP, linearPntT, linearPntB, top, bottom, canvas, midyTolerant, arcTop_right, true);
  64.  
  65. 		cout << "\nmain ==== flag " << flag << endl;
  66. 		if (flag) {
  67. 			cout << "top = " << top.x << " " << top.y << endl;
  68. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  69. 		}
  70. 		canvas.copyTo(src);
  71. 		//cout << "\nmain ==== i " << endl;
  72. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  73. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  74. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  75. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  76. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  77. 		cv::imshow("main-结果图src", src);
  78. 		cv::namedWindow("main-canvas", cv::RECURS_FILTER);
  79. 		cvSetMouseCallback("main-canvas", on_mouse_singleClick, 0);
  80. 		cv::imshow("main-canvas", canvas);
  81. 	}
  82. 	else
  83. 	{
  84. 		vector<string> path_name;
  85. 		getAllFiles(filePath, path_name);
  86. 		cout << "path_name.size()=" << path_name.size() << endl;
  87.  
  88. 		int number = 1;
  89. 		for (auto i : path_name)
  90. 		{
  91. 			//cout << i.size() << endl;
  92. 			if (i.size() == 7) continue;
  93. 			if (i.find(".log") != std::string::npos) continue;
  94. 			if (i.find(".txt") != std::string::npos) continue;
  95. 			if (i.find("_result") != std::string::npos) continue;
  96. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  97. 			std::string path11 = filePath + "\\" + i;
  98. 			cv::Mat src = cv::imread(path11);
  99. 			//cv::Mat src = cv::imread(filePath + "\\" + i);
  100. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  101. 			src.copyTo(canvas);
  102. 			cv::imshow("main_OriginalIMG", src);
  103.  
  104. 			//将结果图写入文件方便查看结果
  105. 			int len = filePath.length();
  106. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  107. 			cout << "i=" << i << ", path11=" << path11 << endl;
  108. 			int time_start = clock();
  109. 			int time_end = clock();
  110.  
  111. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  112. 			//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  113. 			if (getROI) ROI = getROIFromString(i);
  114.  
  115.  
  116. 			bool flag = m_ImageProcessing.LaserParallelToTheMainD_ROI(src, ROI, outMidY, midP, linearPntT, linearPntB, top, bottom, canvas, midyTolerant, arcTop_right, false);
  117.  
  118.  
  119. 			time_end = clock();
  120. 			printf("\n耗时 %d ms \n", time_end - time_start);
  121. 			cout << "\nmain ==== flag " << flag << endl;
  122. 			if (flag) {
  123. 				cout << "top = " << top.x << " " << top.y << endl;
  124. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  125. 			}
  126. 			canvas.copyTo(src);
  127. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  128. 			cv::imshow("main-结果图src", src);
  129. 			cv::imshow("main-canvas", canvas);
  130.  
  131.  
  132. 			//将结果图写入文件方便查看结果
  133. 			char*  filePath_ = new char[len + 1];
  134. 			strcpy(filePath_, filePath.c_str());
  135.  
  136. 			len = filePath.length();
  137. 			char*  resPath_ = new char[len + 1];
  138. 			strcpy(resPath_, resPath.c_str());
  139. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  140.  
  141. 			string out = filePath + resPath;
  142. 			cout << "\nmain ==== i " << i << endl;
  143. 			out.append("\\").append(i);
  144.  
  145. 			cout << "number【" << number << "】res path=" << out << endl;
  146. 			cv::imwrite(out, src);
  147. 			number++;
  148.  
  149. 			cout << "path_name.size() = " << path_name.size() << endl;
  150. 			cout << "*********************************************************************************************************************" << endl;
  151. 			cv::waitKey(10);
  152. 		}
  153. 		cout << "循环结束！！！ main " << endl;
  154. 	}
  155.  
  156. 	ofs.close();
  157. 	cv::waitKey(0);
  158. 	system("Pause");
  159. 	return 0;
  160. }

四点

返回四点+1个点集 

  1. 	/*
  2. 	//矩形位置  返回四点+骨干点集
  3. 	cv::Mat img, 原图
  4. 	cv::Rect ROI,
  5. 	cv::Point & p1, 图像较上方的主管底部点
  6. 	cv::Point & p2, 图像较上方的矩形顶部边缘点
  7. 	cv::Point & p3, 图像较下方的主管底部点
  8. 	cv::Point & p4，图像较下方的矩形顶部边缘点
  9. 	std::vector<cv::Point>& allPntList：直线部分非弧线的点集
  10. 	cv::Mat &canvas,效果图可视化
  11. 	*/
  12. 	bool rectangle4Pnt(cv::Mat imgOriginal, cv::Point & p1, cv::Point& p2, cv::Point & p3, cv::Point& p4, std::vector<cv::Point>& allPntList, cv::Mat &canvas, bool debug = false);
  13. 	bool rectangle4Pnt_ROI(cv::Mat img, cv::Rect ROI, cv::Point & p1, cv::Point& p2, cv::Point & p3, cv::Point& p4, std::vector<cv::Point>& allPntList, cv::Mat &canvas, bool debug = false);
  1. bool PTank_imgProcess::rectangle4Pnt_ROI(cv::Mat img, cv::Rect ROI, cv::Point & p1, cv::Point& p2, cv::Point & p3, cv::Point& p4, std::vector<cv::Point>& allPntList, cv::Mat & canvas, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	p1 = cv::Point(0, 0);
  7. 	p2 = cv::Point(0, 0);
  8. 	p3 = cv::Point(0, 0);
  9. 	p4 = cv::Point(0, 0);
  10. 	allPntList.clear();
  11.  
  12. 	cv::Mat imgOriginal;
  13. 	bool ROI_flag;//是否有给信息区
  14. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  15. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  16. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  17. 	{
  18. 		img.copyTo(imgOriginal);
  19. 		ROI_flag = false;
  20. 	}
  21. 	else
  22. 	{
  23. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  24. 		{
  25. 			printf("信息区两点疑似传反！");
  26. 			return false;
  27. 		}
  28. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  29. 		{
  30. 			printf("信息区的框选超出图像范围！");
  31. 			return false;
  32. 		}
  33. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  34. 		{
  35. 			printf("信息区大小超出图像大小！");
  36. 			return false;
  37. 		}
  38.  
  39. 		//此时不合法的ROI都已提前返回false
  40.  
  41. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  42. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  43. 		ROI_flag = true;
  44. 	}
  45.  
  46. 	result = rectangle4Pnt(imgOriginal, p1, p2, p3, p4, allPntList, canvas, debug);//调用识别算法
  47.  
  48.  
  49. 	if (debug) printf("-------------------------------------****------ roiF  p1=【%d, %d】, p2=【%d, %d】\n", p1.x, p1.y, p2.x, p2.y);
  50. 	if (debug) printf("-------------------------------------****------ roiF  p3=【%d, %d】, p4=【%d, %d】\n", p3.x, p3.y, p4.x, p4.y);
  51. 	//img.copyTo(canvas);
  52.  
  53. 	img.copyTo(canvas);//调试时注释掉
  54. 	if (result)
  55. 	{
  56. 		endRes = 1;
  57. 		p1 += ROI_tl;
  58. 		p2 += ROI_tl;
  59. 		p3 += ROI_tl;
  60. 		p4 += ROI_tl;
  61. 		for (int i = 0; i < allPntList.size(); i++) {
  62. 			allPntList.at(i) += ROI_tl;
  63. 		}
  64. 		int w = 5;
  65. 		for (int i = 0; i < allPntList.size(); i++) {
  66. 			cv::Point curPoint = allPntList.at(i);
  67. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(55, 195, 40), 1);
  68. 		}
  69. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  70. 		circle(canvas, p1, 3, cv::Scalar(0, 69, 255), 1);
  71. 		circle(canvas, p1, 63, cv::Scalar(0, 169, 255), 1);
  72. 		circle(canvas, p2, 3, cv::Scalar(0, 69, 255), 1);
  73. 		circle(canvas, p2, 43, cv::Scalar(255, 169, 0), 1);
  74.  
  75. 		circle(canvas, p3, 3, cv::Scalar(0, 69, 255), 1);
  76. 		circle(canvas, p3, 63, cv::Scalar(0, 169, 255), 1);
  77. 		circle(canvas, p4, 3, cv::Scalar(0, 69, 255), 1);
  78. 		circle(canvas, p4, 43, cv::Scalar(255, 169, 0), 1);
  79. 	}
  80. 	else
  81. 	{
  82. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  83. 	}
  84.  
  85.  
  86. 	cv::Mat ROIcanvas;
  87. 	img.copyTo(ROIcanvas);
  88. 	if (debug) circle(ROIcanvas, p1, 3, cv::Scalar(0, 69, 255), 1);
  89. 	if (debug) circle(ROIcanvas, p1, 63, cv::Scalar(0, 169, 255), 1);
  90. 	if (debug) circle(ROIcanvas, p2, 3, cv::Scalar(0, 69, 255), 1);
  91. 	if (debug) circle(ROIcanvas, p2, 63, cv::Scalar(0, 169, 255), 1);
  92.  
  93. 	if (debug) circle(ROIcanvas, p3, 3, cv::Scalar(0, 69, 255), 1);
  94. 	if (debug) circle(ROIcanvas, p3, 63, cv::Scalar(0, 169, 255), 1);
  95. 	if (debug) circle(ROIcanvas, p4, 3, cv::Scalar(0, 69, 255), 1);
  96. 	if (debug) circle(ROIcanvas, p4, 63, cv::Scalar(0, 169, 255), 1);
  97. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  98. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  99. 	if (debug) printf("---------------****------ roiF  p1=【%d, %d】, p2=【%d, %d】\n", p1.x, p1.y, p2.x, p2.y);
  100. 	if (debug) printf("---------------****------ roiF  p3=【%d, %d】, p4=【%d, %d】\n", p3.x, p3.y, p4.x, p4.y);
  101. 	if (debug) imshow("ROI", ROIcanvas);
  102. 	if (debug) imshow("ROI_canvas", canvas);
  103. 	//ROIcanvas.copyTo(canvas);
  104.  
  105. 	return result;
  106. }

调用main：

  1. //矩形位置  返回四点+骨干点集
  2. int main/*Rectangle4Pnt_ROI*/() {
  3. 	bool flag;
  4. 	//std::string filePath = "../../../../Log08141641/TestImg/src";
  5. 	std::string filePath = "E://vsproject//压力罐\\img\\矩形";
  6.  
  7.  
  8. 	string resPath = "/output";
  9. 	//string resPath = filePath + "./output";
  10.  
  11.  
  12. 	cv::Point tlP = cv::Point(500, 292);
  13. 	cv::Point brP = cv::Point(750, 677);
  14. 	//cv::Rect ROI = cv::Rect(0, 0, 0, 0);
  15. 	//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  16. 	//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  17. 	//cv::Rect ROI = cv::Rect(539, 218, 113, 513);
  18. 	cv::Point top(90, 90);
  19. 	cv::Point bottom(90, 90);
  20. 	cv::Point p3(90, 90);
  21. 	cv::Point p4(90, 90);
  22. 	int outMidY;
  23. 	cv::Mat canvas;
  24. 	PTank_imgProcess m_ImageProcessing;
  25. 	ofstream ofs("log.txt", ios::out);
  26. 	if (ofs)
  27. 	{
  28. 		ofs.clear();
  29. 		ofs << "日志启动" << endl;
  30. 	}
  31. 	std::vector<cv::Point> allPntList;
  32. 	vector<string> path_name;
  33.  
  34.  
  35. 	bool   getROI = true;
  36. 	bool   lianxu = false; //循环测试
  37.  
  38. 	if (lianxu)
  39. 	{
  40. 		string name = cv::format("\\Image_20240627144521008.png");
  41. 		string path = filePath + name;
  42. 		cv::Mat src = cv::imread(path);
  43. 		src.copyTo(canvas);
  44. 		cv::namedWindow("main_OriginalIMG", cv::RECURS_FILTER);
  45. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  46. 		cv::imshow("main_OriginalIMG", src);
  47.  
  48. 		//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  49. 		cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  50. 		if (getROI) ROI = getROIFromString(name);
  51.  
  52.  
  53. 		bool flag = m_ImageProcessing.rectangle4Pnt_ROI(src, ROI, top, bottom, p3, p4, allPntList, canvas, true);
  54.  
  55. 		cout << "\nmain ==== flag " << flag << endl;
  56. 		if (flag) {
  57. 			cout << "top = " << top.x << " " << top.y << endl;
  58. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  59. 		}
  60. 		canvas.copyTo(src);
  61. 		//cout << "\nmain ==== i " << endl;
  62. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  63. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  64. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  65. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  66. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  67. 		cv::imshow("main-结果图src", src);
  68. 		cv::namedWindow("main-canvas", cv::RECURS_FILTER);
  69. 		cvSetMouseCallback("main-canvas", on_mouse_singleClick, 0);
  70. 		cv::imshow("main-canvas", canvas);
  71. 	}
  72. 	else
  73. 	{
  74. 		vector<string> path_name;
  75. 		getAllFiles(filePath, path_name);
  76. 		cout << "path_name.size()=" << path_name.size() << endl;
  77.  
  78. 		int number = 1;
  79. 		for (auto i : path_name)
  80. 		{
  81. 			//cout << i.size() << endl;
  82. 			if (i.size() == 7) continue;
  83. 			if (i.find(".log") != std::string::npos) continue;	
  84. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  85. 			std::string path11 = filePath + "\\" + i;
  86. 			cv::Mat src = cv::imread(path11);
  87. 			//cv::Mat src = cv::imread(filePath + "\\" + i);
  88. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  89. 			src.copyTo(canvas);
  90. 			cv::imshow("main_OriginalIMG", src);
  91.  
  92. 			//将结果图写入文件方便查看结果
  93. 			int len = filePath.length();
  94. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  95. 			cout << "i=" << i << ", path11=" << path11 << endl;
  96. 			int time_start = clock();
  97. 			int time_end = clock();
  98.  
  99. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  100. 			//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  101. 			if (getROI) ROI = getROIFromString(i);
  102.  
  103.  
  104. 			bool flag = m_ImageProcessing.rectangle4Pnt_ROI(src, ROI, top, bottom, p3, p4, allPntList, canvas, false);
  105.  
  106.  
  107. 			time_end = clock();
  108. 			printf("\n耗时 %d ms \n", time_end - time_start);
  109. 			cout << "\nmain ==== flag " << flag << endl;
  110. 			if (flag) {
  111. 				cout << "top = " << top.x << " " << top.y << endl;
  112. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  113. 			}
  114. 			canvas.copyTo(src);
  115. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  116. 			cv::imshow("main-结果图src", src);
  117. 			cv::imshow("main-canvas", canvas);
  118.  
  119.  
  120. 			//将结果图写入文件方便查看结果
  121. 			char*  filePath_ = new char[len + 1];
  122. 			strcpy(filePath_, filePath.c_str());
  123.  
  124. 			len = filePath.length();
  125. 			char*  resPath_ = new char[len + 1];
  126. 			strcpy(resPath_, resPath.c_str());
  127. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  128.  
  129. 			string out = filePath + resPath;
  130. 			cout << "\nmain ==== i " << i << endl;
  131. 			out.append("\\").append(i);
  132.  
  133. 			cout << "number【" << number << "】res path=" << out << endl;
  134. 			cv::imwrite(out, src);
  135. 			number++;
  136.  
  137. 			cout << "path_name.size() = " << path_name.size() << endl;
  138. 			cout << "*********************************************************************************************************************" << endl;
  139. 			cv::waitKey(10);
  140. 		}
  141. 		cout << "循环结束！！！ main " << endl;
  142. 	}
  143.  
  144. 	ofs.close();
  145. 	cv::waitKey(0);
  146. 	system("Pause");
  147. 	return 0;
  148. }

返回四点+双点集

  1. 	/*
  2. 	//多层多道焊  圆管表面 & 四边形坡口拼接
  3. 	cv::Mat img, 原图
  4. 	cv::Point& top, 输出焊点（图像上位于较上方的点）
  5. 	cv::Point& bottom, 输出焊点（图像上位于较下方的点）
  6. 	std::vector<cv::Point>& groovePnt：坡口部分包括内侧面的点集
  7. 	cv::Point& valleyT, 位于焊道底部的焊点（图像上位于较上方的点）
  8. 	cv::Point& valleyB, 位于焊道底部的焊点（图像上位于较下方的点）
  9. 	std::vector<cv::Point>& valleyPnt，位于焊道底部的，焊道底部两点之间的点集
  10. 	cv::Mat &canvas,效果图可视化
  11. 	cv::Rect ROI,
  12. 	*/
  13. 	bool multilayerMultichannel4(cv::Mat imgOriginal, cv::Point& top, cv::Point& bottom, std::vector<cv::Point>& groovePnt, cv::Point& valleyT, cv::Point& valleyB, std::vector<cv::Point>& valleyPnt, cv::Mat &canvas, bool debug = false);
  14. 	bool multilayerMultichannel4_ROI(cv::Mat img, cv::Point& top, cv::Point& bottom, std::vector<cv::Point>& groovePnt, cv::Point& valleyT, cv::Point& valleyB, std::vector<cv::Point>& valleyPnt, cv::Mat &canvas, cv::Rect ROI = cv::Rect(540, 540, 200, 200), bool debug = false);
  1. bool ImageProcess::multilayerMultichannel4_ROI(cv::Mat img, cv::Point & top, cv::Point & bottom, std::vector<cv::Point>& groovePnt, cv::Point & valleyT, cv::Point & valleyB, std::vector<cv::Point>& valleyPnt, cv::Mat & canvas, cv::Rect ROI, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	top = cv::Point(0, 0);
  7. 	bottom = cv::Point(0, 0);
  8. 	groovePnt.clear();
  9. 	valleyT = cv::Point(0, 0);
  10. 	valleyB = cv::Point(0, 0);
  11. 	valleyPnt.clear();
  12.  
  13. 	cv::Mat imgOriginal;
  14. 	bool ROI_flag;//是否有给信息区
  15. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  16. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  17. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  18. 	{
  19. 		img.copyTo(imgOriginal);
  20. 		ROI_flag = false;
  21. 	}
  22. 	else
  23. 	{
  24. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  25. 		{
  26. 			printf("信息区两点疑似传反！");
  27. 			return false;
  28. 		}
  29. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  30. 		{
  31. 			printf("信息区的框选超出图像范围！");
  32. 			return false;
  33. 		}
  34. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  35. 		{
  36. 			printf("信息区大小超出图像大小！");
  37. 			return false;
  38. 		}
  39.  
  40. 		//此时不合法的ROI都已提前返回false
  41.  
  42. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  43. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  44. 		ROI_flag = true;
  45. 	}
  46.  
  47. 	result = multilayerMultichannel4(imgOriginal, top, bottom, groovePnt, valleyT, valleyB, valleyPnt, canvas, debug);//调用识别算法
  48.  
  49.  
  50. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  51. 	if (debug) printf("----- roiF valleyT=【%d, %d】, valleyB=【%d, %d】\n", valleyT.x, valleyT.y, valleyB.x, valleyB.y);
  52. 	//img.copyTo(canvas);
  53.  
  54.  
  55. 	x值和y值补上ROI左上点x和y
  56. 	//if (ROI_flag) {
  57. 	//	if (Corner != cv::Point(0, 0))
  58. 	//	{
  59. 	//		Corner.x += ROI_tl.x;
  60. 	//		Corner.y += ROI_tl.y;
  61. 	//	}
  62. 	//}
  63.  
  64. 	img.copyTo(canvas);//调试时注释掉
  65.  
  66. 	cv::Mat t1;
  67. 	canvas.convertTo(t1, -1, 1900 / 100.0, 100 - 100);//第一次线性变换100
  68. 	if (debug)cv::imshow("线性变换ROI", t1);
  69. 	t1.copyTo(canvas);
  70.  
  71. 	if (result)
  72. 	{
  73. 		endRes = 1;
  74. 		top += ROI_tl;
  75. 		bottom += ROI_tl;
  76. 		valleyT += ROI_tl;
  77. 		valleyB += ROI_tl;
  78. 		for (int i = 0; i < groovePnt.size(); i++) {
  79. 			groovePnt.at(i) += ROI_tl;
  80. 		}
  81. 		for (int i = 0; i < valleyPnt.size(); i++) {
  82. 			valleyPnt.at(i) += ROI_tl;
  83. 		}
  84. 		int w = 9;
  85. 		for (int i = 0; i < groovePnt.size(); i++) {
  86. 			cv::Point curPoint = groovePnt.at(i);
  87. 			cv::line(canvas, curPoint, cv::Point(curPoint.x + w, curPoint.y), cv::Scalar(55, 195, 40), 1);
  88. 		}
  89. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  90. 		circle(canvas, top, 3, cv::Scalar(0, 69, 255), 1);
  91. 		circle(canvas, top, 63, cv::Scalar(0, 169, 255), 1);
  92. 		circle(canvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  93. 		circle(canvas, bottom, 43, cv::Scalar(255, 169, 0), 1);
  94.  
  95. 		circle(canvas, valleyT, 3, cv::Scalar(0, 69, 255), 1);
  96. 		circle(canvas, valleyT, 43, cv::Scalar(0, 169, 255), 1);
  97. 		circle(canvas, valleyB, 3, cv::Scalar(0, 69, 255), 1);
  98. 		circle(canvas, valleyB, 23, cv::Scalar(255, 169, 0), 1);
  99. 		for (int i = 0; i < valleyPnt.size(); i++) {
  100. 			cv::Point curPoint = valleyPnt.at(i);
  101. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(75, 155, 40), 1);
  102. 		}
  103. 	}
  104. 	else
  105. 	{
  106. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  107. 	}
  108.  
  109.  
  110. 	cv::Mat ROIcanvas;
  111. 	img.copyTo(ROIcanvas);
  112. 	if (debug) circle(ROIcanvas, top, 3, cv::Scalar(0, 69, 255), 1);
  113. 	if (debug) circle(ROIcanvas, top, 63, cv::Scalar(0, 169, 255), 1);
  114. 	if (debug) circle(ROIcanvas, bottom, 3, cv::Scalar(0, 69, 255), 1);
  115. 	if (debug) circle(ROIcanvas, bottom, 63, cv::Scalar(0, 169, 255), 1);
  116. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  117. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  118. 	if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  119. 	if (debug) printf("----- roiF valleyT=【%d, %d】, valleyB=【%d, %d】\n", valleyT.x, valleyT.y, valleyB.x, valleyB.y);
  120. 	if (debug) imshow("ROI", ROIcanvas);
  121. 	if (debug) imshow("ROI_canvas", canvas);
  122. 	//ROIcanvas.copyTo(canvas);
  123.  
  124. 	return result;
  125. }

调用main：

  1. //multilayerMultichannel4_ROI  多层多道焊  圆管表面 & 四边形坡口拼接
  2. int main() {
  3. 	bool flag;
  4. 	//std::string filePath = "E:\\vsproject\\WeldingLine\\多层多道4";
  5. 	std::string filePath = "E:\\vsproject\\WeldingLine\\多层多道4\\TestImgAAA";
  6.  
  7.  
  8. 	std::string resPath = "/output";
  9. 	//string resPath = filePath + "./output";
  10.  
  11.  
  12. 	//cv::Point tlP = cv::Point(500, 40);
  13. 	//cv::Point brP = cv::Point(750, 777);
  14. 	cv::Point tlP = cv::Point(345, 108);
  15. 	cv::Point brP = cv::Point(859, 951);
  16. 	cv::Rect ROI = cv::Rect(0, 0, 0, 0);
  17. 	//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  18. 	//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  19. 	//cv::Rect ROI = cv::Rect(544,208,107,545);
  20. 	//cv::Rect ROI = cv::Rect(639, 188, 125, 569);
  21. 	cv::Point top(90, 90);
  22. 	cv::Point bottom(90, 90);
  23. 	cv::Point valleyT(90, 90);
  24. 	cv::Point valleyB(90, 90);
  25. 	int outMidY;
  26. 	cv::Mat canvas;
  27. 	VisualInterface m_ImageProcessing;
  28. 	//ImageProcess m_ImageProcessing;
  29. 	std::vector<cv::Point> groovePnt;
  30. 	std::vector<cv::Point> valleyPnt;
  31. 	vector<string> path_name;
  32. 	bool arcTop_right = true;
  33. 	int midyTolerant = 4;
  34.  
  35. 	bool   getROI = !true;
  36. 	bool   lianxu = false; //循环测试
  37.  
  38. 	if (!lianxu)
  39. 	{
  40. 		string name = cv::format("\\SrcImg_2023_12_28_14_57_17_133_ROI_1132_999_401_126_src.png");
  41. 		//中间底部较高的情况
  42. 		string path = filePath + name;
  43. 		cv::Mat src = cv::imread(path);
  44. 		src.copyTo(canvas);
  45. 		cv::namedWindow("main_OriginalIMG", cv::NORMCONV_FILTER);
  46. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  47. 		cv::imshow("main_OriginalIMG", src);
  48.  
  49. 		cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  50. 		//ROI = cv::Rect(400, 0, 400, src.rows);
  51. 		//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  52. 		if (getROI) ROI = getROIFromString(name);
  53.  
  54.  
  55. 		bool flag = m_ImageProcessing.multilayerMultichannel4_ROI(src, top, bottom, groovePnt, valleyT, valleyB, valleyPnt, canvas, ROI, true);
  56.  
  57. 		cout << "\nmain ==== flag " << flag << endl;
  58. 		if (flag) {
  59. 			cout << "top = " << top.x << " " << top.y << endl;
  60. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  61. 		}
  62. 		canvas.copyTo(src);
  63. 		//cout << "\nmain ==== i " << endl;
  64. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  65. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  66. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  67. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  68. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  69. 		cv::imshow("main-结果图src", src);
  70. 		cv::namedWindow("main-canvas", cv::RECURS_FILTER);
  71. 		cvSetMouseCallback("main-canvas", on_mouse_singleClick, 0);
  72. 		cv::imshow("main-canvas", canvas);
  73. 	}
  74. 	else
  75. 	{
  76. 		vector<string> path_name;
  77. 		getAllFiles(filePath, path_name);
  78. 		cout << "path_name.size()=" << path_name.size() << endl;
  79.  
  80. 		int number = 1;
  81. 		for (auto i : path_name)
  82. 		{
  83. 			//cout << i.size() << endl;
  84. 			if (i.size() == 7) continue;
  85. 			if (i.find(".log") != std::string::npos) continue;
  86. 			if (i.find(".txt") != std::string::npos) continue;
  87. 			if (i.find("_result") != std::string::npos) continue;
  88. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  89. 			std::string path11 = filePath + "\\" + i;
  90. 			cv::Mat src = cv::imread(path11);
  91. 			//cv::Mat src = cv::imread(filePath + "\\" + i);
  92. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  93. 			src.copyTo(canvas);
  94. 			cv::imshow("main_OriginalIMG", src);
  95.  
  96. 			//将结果图写入文件方便查看结果
  97. 			int len = filePath.length();
  98. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  99. 			cout << "i=" << i << ", path11=" << path11 << endl;
  100. 			int time_start = clock();
  101. 			int time_end = clock();
  102.  
  103. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  104. 			//ROI = cv::Rect(400, 0, 400, src.rows);
  105. 			cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  106. 			if (getROI) ROI = getROIFromString(i);
  107.  
  108.  
  109. 			bool flag = m_ImageProcessing.multilayerMultichannel4_ROI(src, top, bottom, groovePnt, valleyT, valleyB, valleyPnt, canvas, ROI, false);
  110.  
  111.  
  112. 			time_end = clock();
  113. 			printf("\n耗时 %d ms \n", time_end - time_start);
  114. 			cout << "\nmain ==== flag " << flag << endl;
  115. 			if (flag) {
  116. 				cout << "top = " << top.x << " " << top.y << endl;
  117. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  118. 			}
  119. 			canvas.copyTo(src);
  120. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  121. 			cv::imshow("main-结果图src", src);
  122. 			cv::imshow("main-canvas", canvas);
  123.  
  124.  
  125. 			//将结果图写入文件方便查看结果
  126. 			char*  filePath_ = new char[len + 1];
  127. 			strcpy(filePath_, filePath.c_str());
  128.  
  129. 			len = filePath.length();
  130. 			char*  resPath_ = new char[len + 1];
  131. 			strcpy(resPath_, resPath.c_str());
  132. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  133.  
  134. 			string out = filePath + resPath;
  135. 			cout << "\nmain ==== i " << i << endl;
  136. 			out.append("\\").append(i);
  137.  
  138. 			cout << "number【" << number << "】res path=" << out << endl;
  139. 			cv::imwrite(out, src);
  140. 			number++;
  141.  
  142. 			cout << "path_name.size() = " << path_name.size() << endl;
  143. 			cout << "*********************************************************************************************************************" << endl;
  144. 			cv::waitKey(10);
  145. 		}
  146. 		cout << "循环结束！！！ main " << endl;
  147. 	}
  148.  
  149. 	cv::waitKey(0);
  150. 	system("Pause");
  151. 	return 0;
  152. }

五点

返回五点+1个点集

  1. 	/*
  2. 	//两弧线夹悬空的圆弧  返回五点+骨干点集   定"L"型管相贯线水平用
  3. 	cv::Mat img, 原图
  4. 	cv::Rect ROI,
  5. 	cv::Point & p1, 图像较上方的主管底部点
  6. 	cv::Point & p2, 图像较上方的弧形激光边缘点
  7. 	cv::Point2f & midP, 圆弧中点
  8. 	cv::Point & p3, 图像较下方的弧形激光边缘点
  9. 	cv::Point & p4，图像较下方的主管底部点
  10. 	std::vector<cv::Point>& allPntList：直线部分非弧线的点集
  11. 	cv::Mat &canvas,效果图可视化
  12. 	*/
  13. 	bool twoArcsClipTheHangingArc(cv::Mat imgOriginal, cv::Point & p1, cv::Point& p2, cv::Point2f& midP, cv::Point& p3, cv::Point& p4, std::vector<cv::Point>& allPntList, cv::Mat &canvas, bool debug = false);
  14. 	bool twoArcsClipTheHangingArc_ROI(cv::Mat img, cv::Rect ROI, cv::Point& p1, cv::Point& p2, cv::Point2f& midP, cv::Point& p3, cv::Point& p4, std::vector<cv::Point>& allPntList, cv::Mat &canvas, bool debug = false);
  1. bool PTank_imgProcess::twoArcsClipTheHangingArc_ROI(cv::Mat img, cv::Rect ROI, cv::Point & p1, cv::Point & p2, cv::Point2f & midP, cv::Point & p3, cv::Point & p4, std::vector<cv::Point>& allPntList, cv::Mat & canvas, bool debug)
  2. {
  3. 	int endRes = 0;
  4. 	bool result = false;
  5. 	img.copyTo(canvas);
  6. 	p1 = cv::Point(0, 0);
  7. 	p2 = cv::Point(0, 0);
  8. 	p3 = cv::Point(0, 0);
  9. 	p4 = cv::Point(0, 0);
  10. 	midP = cv::Point2f(0, 0);
  11. 	allPntList.clear();
  12.  
  13. 	cv::Mat imgOriginal;
  14. 	bool ROI_flag;//是否有给信息区
  15. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  16. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  17. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  18. 	{
  19. 		img.copyTo(imgOriginal);
  20. 		ROI_flag = false;
  21. 	}
  22. 	else
  23. 	{
  24. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  25. 		{
  26. 			printf("信息区两点疑似传反！");
  27. 			return false;
  28. 		}
  29. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  30. 		{
  31. 			printf("信息区的框选超出图像范围！");
  32. 			return false;
  33. 		}
  34. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  35. 		{
  36. 			printf("信息区大小超出图像大小！");
  37. 			return false;
  38. 		}
  39.  
  40. 		//此时不合法的ROI都已提前返回false
  41.  
  42. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  43. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  44. 		ROI_flag = true;
  45. 	}
  46.  
  47. 	result = twoArcsClipTheHangingArc(imgOriginal, p1, p2, midP, p3, p4, allPntList, canvas, debug);//调用识别算法
  48.  
  49.  
  50. 	if (debug) printf("-------------------------------------****------ roiF  p1=【%d, %d】, p2=【%d, %d】\n", p1.x, p1.y, p2.x, p2.y);
  51. 	if (debug) printf("-------------------------------------****------ roiF  p3=【%d, %d】, p4=【%d, %d】\n", p3.x, p3.y, p4.x, p4.y);
  52. 	if (debug) printf("                                                roiF  midP=【%f, %f】\n", midP.x, midP.y);
  53. 	//img.copyTo(canvas);
  54.  
  55. 	img.copyTo(canvas);//调试时注释掉
  56. 	if (result)
  57. 	{
  58. 		endRes = 1;
  59. 		p1 += ROI_tl;
  60. 		p2 += ROI_tl;
  61. 		p3 += ROI_tl;
  62. 		p4 += ROI_tl;
  63. 		midP += cv::Point2f(ROI_tl);
  64. 		for (int i = 0; i < allPntList.size(); i++) {
  65. 			allPntList.at(i) += ROI_tl;
  66. 		}
  67. 		int w = 5;
  68. 		for (int i = 0; i < allPntList.size(); i++) {
  69. 			cv::Point curPoint = allPntList.at(i);
  70. 			cv::line(canvas, curPoint, cv::Point(curPoint.x - w, curPoint.y), cv::Scalar(55, 195, 40), 1);
  71. 		}
  72. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  73. 		circle(canvas, midP, 3, cv::Scalar(0, 69, 255), 1);
  74. 		circle(canvas, midP, 43, cv::Scalar(0, 169, 255), 1);
  75. 		cv::line(canvas, midP, cv::Point2f(5, midP.y), cv::Scalar(155, 195, 40), 1);
  76. 		cv::line(canvas, cv::Point2f(midP.x, midP.y - 5), cv::Point2f(midP.x, midP.y + 5), cv::Scalar(155, 195, 40), 1);
  77.  
  78. 		circle(canvas, p1, 3, cv::Scalar(0, 69, 255), 1);
  79. 		circle(canvas, p1, 63, cv::Scalar(0, 169, 255), 1);
  80. 		circle(canvas, p2, 3, cv::Scalar(0, 69, 255), 1);
  81. 		circle(canvas, p2, 43, cv::Scalar(255, 169, 0), 1);
  82.  
  83. 		circle(canvas, p3, 3, cv::Scalar(0, 69, 255), 1);
  84. 		circle(canvas, p3, 63, cv::Scalar(0, 169, 255), 1);
  85. 		circle(canvas, p4, 3, cv::Scalar(0, 69, 255), 1);
  86. 		circle(canvas, p4, 43, cv::Scalar(255, 169, 0), 1);
  87. 	}
  88. 	else
  89. 	{
  90. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  91. 	}
  92.  
  93.  
  94. 	cv::Mat ROIcanvas;
  95. 	img.copyTo(ROIcanvas);
  96. 	if (debug) circle(ROIcanvas, p1, 3, cv::Scalar(0, 69, 255), 1);
  97. 	if (debug) circle(ROIcanvas, p1, 63, cv::Scalar(0, 169, 255), 1);
  98. 	if (debug) circle(ROIcanvas, p2, 3, cv::Scalar(0, 69, 255), 1);
  99. 	if (debug) circle(ROIcanvas, p2, 63, cv::Scalar(0, 169, 255), 1);
  100.  
  101. 	if (debug) circle(ROIcanvas, p3, 3, cv::Scalar(0, 69, 255), 1);
  102. 	if (debug) circle(ROIcanvas, p3, 63, cv::Scalar(0, 169, 255), 1);
  103. 	if (debug) circle(ROIcanvas, p4, 3, cv::Scalar(0, 69, 255), 1);
  104. 	if (debug) circle(ROIcanvas, p4, 63, cv::Scalar(0, 169, 255), 1);
  105. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  106. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  107. 	if (debug) printf("---------------****------ roiF  p1=【%d, %d】, p2=【%d, %d】\n", p1.x, p1.y, p2.x, p2.y);
  108. 	if (debug) printf("---------------****------ roiF  p3=【%d, %d】, p4=【%d, %d】\n", p3.x, p3.y, p4.x, p4.y);
  109. 	if (debug) printf("---------------****------ roiF  midP=【%f, %f】\n", midP.x, midP.y);
  110. 	if (debug) imshow("ROI", ROIcanvas);
  111. 	if (debug) imshow("ROI_canvas", canvas);
  112. 	//ROIcanvas.copyTo(canvas);
  113.  
  114. 	return result;
  115. }

 调用main：

  1. //两弧线夹悬空的圆弧  返回五点+骨干点集   定"L"型管相贯线水平用
  2. int mainTwoArcsClipTheHangingArc() {
  3. 	bool flag;
  4. 	//std::string filePath = "../../../../Log08141641/TestImg/src";
  5. 	std::string filePath = "E://vsproject//压力罐\\img\\L型管定水平";
  6.  
  7.  
  8. 	string resPath = "/output";
  9. 	//string resPath = filePath + "./output";
  10.  
  11.  
  12. 	cv::Point tlP = cv::Point(500, 292);
  13. 	cv::Point brP = cv::Point(750, 677);
  14. 	//cv::Rect ROI = cv::Rect(0, 0, 0, 0);
  15. 	//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  16. 	//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  17. 	//cv::Rect ROI = cv::Rect(539, 218, 113, 513);
  18. 	cv::Point top(90, 90);
  19. 	cv::Point bottom(90, 90);
  20. 	cv::Point p3(90, 90);
  21. 	cv::Point p4(90, 90);
  22. 	cv::Point2f midP(90, 90);
  23. 	int outMidY;
  24. 	cv::Mat canvas;
  25. 	PTank_imgProcess m_ImageProcessing;
  26. 	ofstream ofs("log.txt", ios::out);
  27. 	if (ofs)
  28. 	{
  29. 		ofs.clear();
  30. 		ofs << "日志启动" << endl;
  31. 	}
  32. 	std::vector<cv::Point> allPntList;
  33. 	std::vector<cv::Point> tPntList;
  34. 	std::vector<cv::Point> bPntList;
  35. 	std::vector<cv::Point> mPntList;
  36. 	vector<string> path_name;
  37.  
  38.  
  39. 	bool   getROI = true;
  40. 	bool   lianxu = false; //循环测试
  41.  
  42. 	if (!lianxu)
  43. 	{
  44. 		string name = cv::format("\\Image_20240627151133378.png");
  45. 		string path = filePath + name;
  46. 		cv::Mat src = cv::imread(path);
  47. 		src.copyTo(canvas);
  48. 		cv::namedWindow("main_OriginalIMG", cv::RECURS_FILTER);
  49. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  50. 		cv::imshow("main_OriginalIMG", src);
  51.  
  52. 		//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  53. 		cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  54. 		if (getROI) ROI = getROIFromString(name);
  55.  
  56.  
  57. 		bool flag = m_ImageProcessing.twoArcsClipTheHangingArc_ROI(src, ROI, top, bottom, midP, p3, p4, allPntList, canvas, true);
  58. 		m_ImageProcessing.By4Pnt_SplitPLst_1to3(canvas, top, bottom, p3, p4, allPntList, tPntList, bPntList, mPntList, canvas, true);
  59.  
  60. 		cout << "\nmain ==== flag " << flag << endl;
  61. 		if (flag) {
  62. 			cout << "top = " << top.x << " " << top.y << endl;
  63. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  64. 			cout << "midP = " << midP.x << " " << midP.y << endl;
  65. 		}
  66. 		canvas.copyTo(src);
  67. 		//cout << "\nmain ==== i " << endl;
  68. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  69. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  70. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  71. 		cv::namedWindow("main-结果图src", cv::RECURS_FILTER);
  72. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  73. 		cv::imshow("main-结果图src", src);
  74. 		cv::namedWindow("main-canvas", cv::RECURS_FILTER);
  75. 		cvSetMouseCallback("main-canvas", on_mouse_singleClick, 0);
  76. 		cv::imshow("main-canvas", canvas);
  77. 	}
  78. 	else
  79. 	{
  80. 		vector<string> path_name;
  81. 		getAllFiles(filePath, path_name);
  82. 		cout << "path_name.size()=" << path_name.size() << endl;
  83.  
  84. 		int number = 1;
  85. 		for (auto i : path_name)
  86. 		{
  87. 			//cout << i.size() << endl;
  88. 			if (i.size() == 7) continue;
  89. 			if (i.find(".log") != std::string::npos) continue;
  90. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  91. 			std::string path11 = filePath + "\\" + i;
  92. 			cv::Mat src = cv::imread(path11);
  93. 			//cv::Mat src = cv::imread(filePath + "\\" + i);
  94. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  95. 			src.copyTo(canvas);
  96. 			cv::imshow("main_OriginalIMG", src);
  97.  
  98. 			//将结果图写入文件方便查看结果
  99. 			int len = filePath.length();
  100. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  101. 			cout << "i=" << i << ", path11=" << path11 << endl;
  102. 			int time_start = clock();
  103. 			int time_end = clock();
  104.  
  105. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  106. 			//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  107. 			if (getROI) ROI = getROIFromString(i);
  108.  
  109.  
  110. 			bool flag = m_ImageProcessing.twoArcsClipTheHangingArc_ROI(src, ROI, top, bottom, midP, p3, p4, allPntList, canvas, false);
  111. 			m_ImageProcessing.By4Pnt_SplitPLst_1to3(canvas, top, bottom, p3, p4, allPntList, tPntList, bPntList, mPntList, canvas, false);
  112.  
  113.  
  114. 			time_end = clock();
  115. 			printf("\n耗时 %d ms \n", time_end - time_start);
  116. 			cout << "\nmain ==== flag " << flag << endl;
  117. 			if (flag) {
  118. 				cout << "top = " << top.x << " " << top.y << endl;
  119. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  120. 				cout << "midP = " << midP.x << " " << midP.y << endl;
  121. 			}
  122. 			canvas.copyTo(src);
  123. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  124. 			cv::imshow("main-结果图src", src);
  125. 			cv::imshow("main-canvas", canvas);
  126.  
  127.  
  128. 			//将结果图写入文件方便查看结果
  129. 			char*  filePath_ = new char[len + 1];
  130. 			strcpy(filePath_, filePath.c_str());
  131.  
  132. 			len = filePath.length();
  133. 			char*  resPath_ = new char[len + 1];
  134. 			strcpy(resPath_, resPath.c_str());
  135. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  136.  
  137. 			string out = filePath + resPath;
  138. 			cout << "\nmain ==== i " << i << endl;
  139. 			out.append("\\").append(i);
  140.  
  141. 			cout << "number【" << number << "】res path=" << out << endl;
  142. 			cv::imwrite(out, src);
  143. 			number++;
  144.  
  145. 			cout << "path_name.size() = " << path_name.size() << endl;
  146. 			cout << "*********************************************************************************************************************" << endl;
  147. 			cv::waitKey(10);
  148. 		}
  149. 		cout << "循环结束！！！ main " << endl;
  150. 	}
  151.  
  152. 	ofs.close();
  153. 	cv::waitKey(0);
  154. 	system("Pause");
  155. 	return 0;
  156. }

传入原图和模板图，返回vector

  1. 	/*
  2. 	//泡泡
  3. 	cv::Mat img, 原图
  4. 	cv::Rect ROI, 
  5. 	cv::Mat imgTemplate：模板图片(无泡泡的图)
  6. 	cv::Mat &canvas,效果图可视化
  7. 	*/
  8. 	bool DetectingBubble(const cv::Mat imgOriginal, cv::Mat imgTemplate, std::vector<cv::Rect> &bubbles, cv::Mat &canvas, bool debug = false);
  9. 	bool DetectingBubble_ROI(cv::Mat img, cv::Mat imgTemplate, std::vector<cv::Rect> &bubbles, cv::Rect ROI, cv::Mat &canvas, bool debug = false);
  1. bool ImageProcessing::DetectingBubble_ROI(cv::Mat img, cv::Mat imgTemplate, std::vector<cv::Rect> &bubbles, cv::Rect ROI, cv::Mat & canvas, bool debug)
  2. {
  3. 	cout << __FUNCTION__ << endl;
  4. 	int endRes = 0;
  5. 	bool result = false;
  6. 	img.copyTo(canvas);
  7. 	bubbles.clear();
  8.  
  9. 	cv::Mat imgOriginal;
  10. 	bool ROI_flag;//是否有给信息区
  11. 	cv::Point ROI_tl = cv::Point2i(ROI.x, ROI.y);
  12. 	cv::Point ROI_br = cv::Point2i(ROI.x + ROI.width, ROI.y + ROI.height);
  13. 	if (ROI_tl == cv::Point(0, 0) && ROI_br == cv::Point(0, 0) || ROI.width == 0 || ROI.height == 0)
  14. 	{
  15. 		img.copyTo(imgOriginal);
  16. 		ROI_flag = false;
  17. 	}
  18. 	else
  19. 	{
  20. 		if (ROI_tl.x > ROI_br.x || ROI_tl.y > ROI_br.y)//信息区两点疑似传反
  21. 		{
  22. 			printf("信息区两点疑似传反！");
  23. 			return false;
  24. 		}
  25. 		if (ROI_tl.x < 0 || ROI_br.x > img.cols || ROI_tl.y < 0 || ROI_br.y > img.rows)//信息区的框选超出图像范围
  26. 		{
  27. 			printf("信息区的框选超出图像范围！");
  28. 			return false;
  29. 		}
  30. 		if (abs(ROI_tl.x - ROI_br.x) > img.cols || abs(ROI_tl.y - ROI_br.y) > img.rows)//信息区大小超出图像大小
  31. 		{
  32. 			printf("信息区大小超出图像大小！");
  33. 			return false;
  34. 		}
  35.  
  36. 		//此时不合法的ROI都已提前返回false
  37.  
  38. 		cv::Rect roi(ROI_tl.x, ROI_tl.y, ROI_br.x - ROI_tl.x, ROI_br.y - ROI_tl.y);
  39. 		imgOriginal = img(roi);	//裁剪出的ROI区域放于imgOriginal
  40. 		imgTemplate = imgTemplate(roi);
  41. 		ROI_flag = true;
  42. 	}
  43.  
  44. 	result = DetectingBubble(imgOriginal, imgTemplate, bubbles, canvas, debug);//调用识别算法
  45.  
  46.  
  47. 	//if (debug) printf("-------------------------------------****------ roiF  top=【%d, %d】, bottom=【%d, %d】\n", top.x, top.y, bottom.x, bottom.y);
  48.  
  49.  
  50. 	img.copyTo(canvas);//调试时注释掉
  51.  
  52. 	cv::Mat t1;
  53. 	canvas.convertTo(t1, -1, 200 / 100.0, 0 + 55);
  54. 	//if (debug) cv::imshow("线性变换0", t1);
  55. 	t1.copyTo(canvas);
  56.  
  57. 	if (result)
  58. 	{
  59. 		endRes = 1;
  60. 		cv::putText(canvas, "true", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  61. 		//for (int i = 0; i < bubbles.size(); i++) {
  62. 		//	//bubbles.at(i).tl.x = bubbles.at(i).tl.x + ROI_tl.x;
  63. 		//	//bubbles.at(i).tl.y = bubbles.at(i).tl.y + ROI_tl.y;
  64. 		//	//bubbles.at(i).br.x = bubbles.at(i).br.x + ROI_tl.x;
  65. 		//	//bubbles.at(i).br.y = bubbles.at(i).br.y + ROI_tl.y;
  66. 		//	cv::Rect curBR = bubbles.at(i);
  67. 		//	bubbles.at(i).x = bubbles.at(i).x + ROI_tl.x;
  68. 		//	bubbles.at(i).y = bubbles.at(i).y + ROI_tl.y;
  69. 		//	//curBR.tl.x = bubbles.at(i).tl.x + ROI_tl.x;
  70. 		//	//curBR.tl.y = bubbles.at(i).tl.y + ROI_tl.y;
  71. 		//	//curBR.br.x = bubbles.at(i).br.x + ROI_tl.x;
  72. 		//	//curBR.br.y = bubbles.at(i).br.y + ROI_tl.y;
  73. 		//}
  74. 		for (int i = 0; i < bubbles.size(); i++) {//可视化
  75. 			bubbles.at(i).x = bubbles.at(i).x + ROI_tl.x;
  76. 			bubbles.at(i).y = bubbles.at(i).y + ROI_tl.y;
  77. 			cv::Rect curBR = bubbles.at(i);
  78. 			cv::rectangle(canvas, curBR.tl(), curBR.br(), cv::Scalar(0, 255, 0), 1);
  79. 			cv::putText(canvas, to_string(i), curBR.br(), cv::FONT_HERSHEY_COMPLEX, 0.45, cv::Scalar(55, 35, 160), 1);
  80. 		}
  81. 	}
  82. 	else
  83. 	{
  84. 		cv::putText(canvas, "false", { cv::Point(50, 50) }, cv::FONT_HERSHEY_COMPLEX, 1, cv::Scalar(255, 255, 255), 1);
  85. 	}
  86.  
  87. 	cv::Mat ROIcanvas;
  88. 	img.copyTo(ROIcanvas);
  89. 	if (debug) rectangle(ROIcanvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  90. 	rectangle(canvas, ROI_tl, ROI_br, cv::Scalar(0, 0, 255), 1);
  91. 	if (debug) imshow("ROI", ROIcanvas);
  92. 	if (debug) imshow("ROI_canvas", canvas);
  93. 	//ROIcanvas.copyTo(canvas);
  94.  
  95. 	return result;
  96. }

 调用main：

  1. //泡泡  OneSideSquare_ROI
  2. int main/*pp*/() {
  3. 	bool flag;
  4. 	std::string filePath = "../../../../泡泡/0708/6000";
  5. 	//std::string filePath = "../../../../泡泡/0708/20000";
  6.  
  7.  
  8. 	string resPath = "/output";
  9. 	//string resPath = filePath + "./output";
  10.  
  11.  
  12. 	cv::Point tlP = cv::Point(491, 50);
  13. 	cv::Point brP = cv::Point(1596, 760);
  14. 	tlP = cv::Point(1, 296);
  15. 	brP = cv::Point(1160, 2278);
  16. 	tlP = cv::Point(190, 0);
  17. 	brP = cv::Point(840, 1024);
  18. 	//cv::Rect ROI = cv::Rect(0, 0, 0, 0);
  19. 	//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  20. 	//cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  21. 	//cv::Rect ROI = cv::Rect(639, 188, 125, 569);
  22. 	cv::Point top(90, 90);
  23. 	cv::Point bottom(90, 90);
  24. 	cv::Point2f midP(90, 90);
  25. 	int outMidY;
  26. 	cv::Mat canvas;
  27. 	ImageProcessing m_ImageProcessing;
  28. 	ofstream ofs("log.txt", ios::out);
  29. 	if (ofs)
  30. 	{
  31. 		ofs.clear();
  32. 		ofs << "日志启动" << endl;
  33. 	}
  34. 	std::vector<cv::Rect> bubbles;//泡泡集合
  35. 	vector<Picture_set> img_buf;
  36. 	vector<string> path_name;
  37. 	//cv::Mat templ = cv::imread(filePath + cv::format("\\屏幕截图(13).png"));//模板图的灰度图
  38. 	cv::Mat templ = cv::imread(filePath + cv::format("\\Image_20240708175744959.png"));//6000
  39.  
  40. 	bool   getROI = !true;
  41. 	bool   lianxu = false; //循环测试
  42.  
  43. 	if (lianxu)
  44. 	{
  45. 		std::string name = cv::format("\\Image_20240708175747311.png");
  46. 		string path = filePath + name;
  47. 		cv::Mat src = cv::imread(path);
  48. 		src.copyTo(canvas);
  49. 		cv::namedWindow("main_OriginalIMG", cv::NORMCONV_FILTER);
  50. 		cvSetMouseCallback("main_OriginalIMG", on_mouse_singleClick, 0);
  51. 		cv::imshow("main_OriginalIMG", src);
  52. 		cv::imshow("templ", templ);
  53.  
  54. 		//cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  55. 		cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  56. 		if (getROI) {
  57. 			ROI = getROIFromString(name);
  58. 		}
  59. 		//cv::Mat imgCur = src - templ;
  60. 		imgCur = imgOriginal - imgTemplate;
  61. 		//if (1) cv::imshow("减去模板图后", imgCur);
  62.  
  63. 		bool flag = m_ImageProcessing.DetectingBubble_ROI(src, templ, bubbles, ROI, canvas, true);
  64.  
  65. 		cout << "\nmain ==== flag " << flag << endl;
  66. 		if (flag) {
  67. 			cout << "top = " << top.x << " " << top.y << endl;
  68. 			cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  69. 		}
  70. 		canvas.copyTo(src);
  71. 		//cout << "\nmain ==== i " << endl;
  72. 		circle(canvas, top, 63, cv::Scalar(0, 69, 255), 1);
  73. 		circle(canvas, bottom, 63, cv::Scalar(0, 69, 255), 1);
  74. 		cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  75. 		cv::namedWindow("main-结果图src", cv::NORMCONV_FILTER);
  76. 		cvSetMouseCallback("main-结果图src", on_mouse_singleClick, 0);
  77. 		cv::imshow("main-结果图src", src);
  78. 		cv::namedWindow("main-canvas", cv::NORMCONV_FILTER);
  79. 		cvSetMouseCallback("main-canvas", on_mouse_singleClick, 0);
  80. 		cv::imshow("main-canvas", canvas);
  81. 	}
  82. 	else
  83. 	{
  84. 		vector<string> path_name;
  85. 		getAllFiles(filePath, path_name);
  86. 		cout << "path_name.size()=" << path_name.size() << endl;
  87.  
  88. 		int number = 1;
  89. 		for (auto i : path_name)
  90. 		{
  91. 			//cout << i.size() << endl;
  92. 			if (i.size() == 7) continue;
  93. 			if (i.find("output") != std::string::npos) continue;
  94. 			if (i.find(".log") != std::string::npos) continue;
  95. 			if (i.find(".txt") != std::string::npos) continue;
  96. 			if (i.find("png") == std::string::npos && i.find("bmp") == std::string::npos && i.find("jpg") == std::string::npos) continue;
  97. 			//if (i.find("_202404220930171") == std::string::npos) continue;
  98. 			std::string name = i;
  99. 			std::string path11 = filePath + "\\" + i;
  100. 			cv::Mat src = cv::imread(path11);
  101. 			//cv::Mat src = cv::imread(filePath + "\\" + i);
  102. 			cout << endl << "number【" << number << "】src path=" << i << " ;" << endl;
  103. 			src.copyTo(canvas);
  104. 			cv::imshow("main_OriginalIMG", src);
  105.  
  106. 			//将结果图写入文件方便查看结果
  107. 			int len = filePath.length();
  108. 			//i = i.substr(i.find_last_of('\\') + 1, len);
  109. 			cout << "i=" << i << ", path11=" << path11 << endl;
  110. 			int time_start = clock();
  111. 			int time_end = clock();
  112.  
  113. 			cv::Rect ROI = cv::Rect(0, 0, src.cols, src.rows);
  114. 			cv::Rect ROI = cv::Rect(tlP.x, tlP.y, brP.x - tlP.x, brP.y - tlP.y);
  115. 			if (getROI) {
  116. 				ROI = getROIFromString(name);
  117. 			}
  118.  
  119. 			bool flag = m_ImageProcessing.DetectingBubble_ROI(src, templ, bubbles, ROI, canvas, false);
  120.  
  121.  
  122. 			time_end = clock();
  123. 			printf("\n耗时 %d ms \n", time_end - time_start);
  124. 			cout << "\nmain ==== flag " << flag << endl;
  125. 			if (flag) {
  126. 				cout << "top = " << top.x << " " << top.y << endl;
  127. 				cout << "bottom = " << bottom.x << " " << bottom.y << endl;
  128. 			}
  129. 			canvas.copyTo(src);
  130. 			//cv::putText(src, to_string(number - 1), { cv::Point(250, 250) }, cv::FONT_HERSHEY_COMPLEX, 8, cv::Scalar(255, 255, 255), 5);
  131. 			cv::putText(src, to_string(flag), cv::Point(src.cols / 2 - 20, src.rows / 2 - 65), cv::FONT_HERSHEY_COMPLEX, 1.75, cv::Scalar(255, 135, 160), 3);
  132. 			cv::namedWindow("main-结果图src", cv::NORMCONV_FILTER);
  133. 			cv::imshow("main-结果图src", src);
  134. 			cv::namedWindow("main-canvas", cv::NORMCONV_FILTER);
  135. 			cv::imshow("main-canvas", canvas);
  136.  
  137.  
  138. 			//将结果图写入文件方便查看结果
  139. 			char*  filePath_ = new char[len + 1];
  140. 			strcpy(filePath_, filePath.c_str());
  141.  
  142. 			len = filePath.length();
  143. 			char*  resPath_ = new char[len + 1];
  144. 			strcpy(resPath_, resPath.c_str());
  145. 			//i.erase(0, strlen(filePath_) - strlen(resPath_) - 2);
  146.  
  147. 			string out = filePath + resPath;
  148. 			cout << "\nmain ==== i " << i << endl;
  149. 			out.append("\\").append(i);
  150.  
  151. 			cout << "number【" << number << "】res path=" << out << endl;
  152. 			cv::imwrite(out, src);
  153. 			number++;
  154.  
  155. 			cout << "path_name.size() = " << path_name.size() << endl;
  156. 			cout << "*********************************************************************************************************************" << endl;
  157. 			cv::waitKey(1);
  158. 		}
  159. 		cout << "循环结束！！！ main " << endl;
  160. 	}
  161.  
  162. 	ofs.close();
  163. 	cv::waitKey(0);
  164. 	system("Pause");
  165. 	return 0;
  166. }

***识别接口开头模板***

  1. {
  2. 	bool result = false;
  3. 	if (imgOriginal.empty()) {
  4. 		printf("传入图像为空，直接返回false！！！！ \n");
  5. 		return false;
  6. 	}
  7. 	imgOriginal.copyTo(canvas);
  8. 	top = cv::Point(0, 0);
  9. 	bottom = cv::Point(0, 0);
  10. 	midP = cv::Point(0, 0);
  11.  
  12. 	cv::Mat imgGray;
  13. 	if (imgOriginal.type() != CV_8UC1)
  14. 		cv::cvtColor(imgOriginal, imgGray, cv::COLOR_BGR2GRAY);
  15. 	else imgOriginal.copyTo(imgGray);
  16.  
  17. 	//if (1) {
  18. 	//	cv::Mat imgMedianBlur;
  19. 	//	cv::medianBlur(imgGray, imgMedianBlur, 3);
  20. 	//	if (debug) cv::imshow("imgMedianBlur", imgMedianBlur);
  21. 	//	imgMedianBlur.copyTo(imgGray);
  22. 	//}
  23.  
  24. 	cv::Mat close;
  25. 	cv::Mat kernel = cv::getStructuringElement(cv::MORPH_RECT, cv::Size(7, 5));
  26. 	cv::morphologyEx(imgGray, close, cv::MORPH_CLOSE, kernel);
  27. 	if (debug) cv::imshow("闭运算", close);
  28.  
  29. 	cv::Mat thresholdMat;
  30. 	cv::adaptiveThreshold(close, thresholdMat, 255, cv::ADAPTIVE_THRESH_GAUSSIAN_C, cv::THRESH_BINARY, 85, -15);
  31. 	if (debug) cv::imshow("threshold", thresholdMat);
  32.  
  33. 	//一个不为0的像素都没有，即全黑时，直接返回false
  34. 	int iVal255 = cv::countNonZero(thresholdMat);
  35. 	if (!iVal255) return false;
  36.  
  37.  
  38. 	std::vector<std::vector<cv::Point>> contours;
  39. 	std::vector<cv::Vec4i> hierarchy;
  40. 	std::vector<std::vector<cv::Point>>::iterator k;
  41. 	cv::findContours(thresholdMat, contours, hierarchy, cv::RETR_EXTERNAL, CV_RETR_LIST);
  42.  
  43. 	if (contours.size() <= 0)//图为全黑时
  44. 	{
  45. 		std::cout << "contours.size() <= 0";
  46. 		return false;
  47. 	}
  48.  
  49. 	if (debug) printf("图像处理后 检测到的轮廓数 contours.size() = %d  \n", contours.size());
  50. 	int remainNum = 0;//剩余的轮廓数
  51. 	cv::Mat  tmpMat = cv::Mat(thresholdMat.rows, thresholdMat.cols, CV_8UC1, cv::Scalar(0, 0, 0));
  52. 	//画出轮廓;
  53. 	int  count = 0;
  54. 	for (k = contours.begin(); k != contours.end(); ++k, count++) //删除小连通域的
  55. 	{
  56. 		std::vector<cv::Point> curContours = *k;
  57. 		if (curContours.size() < 40) continue;
  58. 		remainNum++;
  59. 		cv::drawContours(tmpMat, contours, count, cv::Scalar(255, 255, 255), -1, CV_AA, hierarchy); //用全黑色填充
  60. 	}
  61. 	if (debug) cv::imshow("tmpMat", tmpMat);
  62. 	if (debug) printf("删减后 剩余的轮廓数 remainNum = %d \n", remainNum);
  63.  
  64. 	if (remainNum == 0)//图为全黑时
  65. 	{
  66. 		std::cout << "contours.size() <= 0";
  67. 		return false;
  68. 	}
  69.  
  70. 	//------------------------------------------------------------------------------------------------------------------------------------
  71. 	//提取骨干点集
  72.  
  73. 	std::vector<cv::Point> bright_point;//max点
  74. 	std::vector<int> bright_value;//max像素值
  75. 	//bright_point = extrectBackbone(tmpMat, bright_value);
  76. 	if (1) {
  77. 		if (debug) cout << "tmpMat.size() =" << tmpMat.size() << std::endl;
  78. 		int splitRow = tmpMat.rows / 2 - 5;
  79. 		cv::Mat tmpMatU, tmpMatD;
  80. 		tmpMat.rowRange(0, splitRow).copyTo(tmpMatU);
  81. 		tmpMat.rowRange(splitRow, tmpMat.rows).copyTo(tmpMatD);
  82.  
  83. 		cv::flip(tmpMatU, tmpMatU, 0);
  84.  
  85. 		std::vector<cv::Point> bright_pointU;//max点
  86. 		std::vector<cv::Point> bright_pointD;//max点
  87. 		std::vector<int> bright_valueU;//max像素值
  88. 		std::vector<int> bright_valueD;//max像素值
  89. 		bright_pointU = extrectBackbone(tmpMatU, bright_valueU, 0);
  90. 		bright_pointD = extrectBackbone(tmpMatD, bright_valueD, 0);
  91.  
  92. 		std::reverse(bright_pointU.begin(), bright_pointU.end());
  93. 		for (int i = 0; i < bright_pointU.size(); i++) {//把上半部分的骨干上下翻转回
  94. 			cv::Point curPoint = bright_pointU.at(i);
  95. 			if (curPoint.x != 0) curPoint.y = tmpMatU.rows - curPoint.y - 1;
  96. 			//if(debug) printf("将上半截骨干排序反转：第%d 行, 【%d, %d】 =============  \n", i, curPoint.x, curPoint.y);
  97. 			//curPoint.x = tmpMat.cols - curPoint.x;
  98. 			bright_point.push_back(curPoint);
  99. 		}
  100. 		std::vector<cv::Point> bright_pointD_new;
  101. 		for (int i = 0; i < bright_pointD.size(); i++) {
  102. 			cv::Point curPoint = bright_pointD.at(i);
  103. 			if (curPoint.x != 0) curPoint.y = splitRow + curPoint.y;
  104. 			bright_pointD_new.push_back(curPoint);
  105. 		}
  106. 		bright_pointD = bright_pointD_new;
  107.  
  108. 		bright_point.insert(bright_point.end(), bright_pointD.begin(), bright_pointD.end());
  109.  
  110. 		std::reverse(bright_valueU.begin(), bright_valueU.end());
  111. 		bright_value = bright_valueU;
  112. 		bright_value.insert(bright_value.end(), bright_valueD.begin(), bright_valueD.end());
  113. 	}
  114.  
  115. 	//有效骨干个数过少，直接返回 false ！！！！
  116. 	int vaildBackbone = 0;//有效骨干个数
  117. 	for (int i = 0; i < bright_point.size(); i++)
  118. 	{
  119. 		cv::Point P = bright_point.at(i);
  120. 		if (P != cv::Point(0, 0)) vaildBackbone++;
  121. 	}
  122. 	if (debug) printf("有效骨干个数 vaildBackbone = %d \n", vaildBackbone);
  123. 	if (vaildBackbone <= 30)
  124. 	{
  125. 		printf("有效骨干个数过少！！！！\n");
  126. 		return false;
  127. 	}
  128.  
  129. 	if (debug) {
  130. 		printf("骨干点集更新前：\n");
  131. 		for (int i = 0; i < bright_point.size(); i++)
  132. 		{
  133. 			cv::Point curPoint = bright_point.at(i);
  134. 			int curGray = bright_value.at(i);
  135. 			//printf("骨干点集更新前：第%d 行, 【%d, %d】 灰度值：%d \n", i, curPoint.x, curPoint.y, curGray);
  136. 		}
  137. 	}
  138.  
  139. 	std::vector<cv::Point> tmpPoint;
  140. 	std::vector<int> tmpValue;
  141. 	for (int i = 0; i < bright_point.size(); i++) {
  142. 		cv::Point curPoint = bright_point.at(i);
  143. 		if (curPoint.x == 0 || curPoint.y == 0) continue;
  144. 		if (tmpMat.at<uchar>(curPoint) < 185) continue;
  145. 		//if (imgGray.at<uchar>(curPoint) < 185) continue;
  146. 		tmpPoint.push_back(curPoint);
  147. 	}
  148. 	bright_point = tmpPoint;//把无效点删掉
  149.  
  150. 	if (0) {
  151. 		std::vector<cv::Point> rightmost;//激光最右点集
  152. 		//for (int row = 0; row < tmpMat.rows; row++)
  153. 		//for (int row = bright_point.at(0).y; row < bright_point.at(bright_point.size() - 1).y; row++)
  154. 		for (int i = 0; i < bright_point.size(); i++)
  155. 		{
  156. 			int row = bright_point[i].y;
  157. 			//for (int col = bright_point.at(row).x; col < tmpMat.cols - 1; col++)//-1是为了防崩溃
  158. 			for (int col = bright_point.at(i).x; col < tmpMat.cols - 1 && col < bright_point.at(i).x + 70; col++)//-1是为了防崩溃
  159. 			{
  160. 				int pv = tmpMat.at<uchar>(row, col);
  161. 				//int pv = tmpMat.at<uchar>(row, col);
  162. 				if (pv != 0 && tmpMat.at<uchar>(row, col + 1) == 0)//激光最右点入列
  163. 				{
  164. 					if (!rightmost.empty() && rightmost.back().y == row) rightmost.pop_back();
  165. 					rightmost.push_back(cv::Point(col - 1, row));
  166. 					//break;
  167. 				}
  168. 			}
  169. 		}
  170. 		bright_point = rightmost;
  171. 	}
  172.  
  173.  
  174. 	if (debug) {
  175. 		printf("骨干点集更新后：\n");
  176. 		for (int i = 0; i < bright_point.size(); i++) {
  177. 			cv::Point curPoint = bright_point.at(i);
  178. 			printf("骨干点集更新后：第%d 行, 【%d, %d】 \n", i, curPoint.x, curPoint.y);
  179. 		}
  180. 	}
  181. 	if (debug) printf("有效骨干个数 bright_point.size() = %d \n", bright_point.size());
  182. 	if (bright_point.size() <= 30)
  183. 	{
  184. 		printf("有效骨干个数过少！！！！\n");
  185. 		return false;
  186. 	}
  187.  
  188. 	allPntList = bright_point;
  189.  
  190. 	//可视化骨干点集
  191. 	cv::Mat newArcMat = cv::Mat::zeros(tmpMat.size(), CV_8UC1);
  192. 	for (int i = 0; i < bright_point.size(); i++)
  193. 	{
  194. 		cv::Point P = cv::Point(bright_point.at(i).x, bright_point.at(i).y);
  195. 		newArcMat.at<uchar>(cv::Point(bright_point.at(i).x, bright_point.at(i).y)) = 255;
  196. 	}
  197. 	cv::Mat BowCurveMat;
  198. 	cv::Mat imgRGB_Begin;
  199. 	if (newArcMat.type() != CV_8UC1)
  200. 		cvtColor(newArcMat, imgRGB_Begin, cv::COLOR_RGB2GRAY);
  201. 	else newArcMat.copyTo(imgRGB_Begin);
  202.  
  203.  
  204. 	//重叠看效果
  205. 	cv::addWeighted(tmpMat, 0.5, imgRGB_Begin, 0.5, 0, BowCurveMat);
  206. 	//if (debug) cv::namedWindow("BowCurveMat", cv::WINDOW_NORMAL);
  207. 	if (debug) cv::imshow("BowCurveMat", BowCurveMat);
  208.  
  209. 	result = true;
  210.  
  211.     
  212.     // TODO
  213.     
  214.  
  215. 	return result;
  216. }

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/运维做开发/article/detail/825594
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号