From 30f434d78b58e3a4198cf5ba5a9e5a0ce1cd5292 Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期四, 07 七月 2022 14:32:36 +0800
Subject: [PATCH] 'bug修复'

---
 ConsoleApplication/ImgUtil.cpp |   18 +++++++++++++++++-
 1 files changed, 17 insertions(+), 1 deletions(-)

diff --git a/ConsoleApplication/ImgUtil.cpp b/ConsoleApplication/ImgUtil.cpp
index 8259bb7..0a9d58a 100644
--- a/ConsoleApplication/ImgUtil.cpp
+++ b/ConsoleApplication/ImgUtil.cpp
@@ -279,6 +279,14 @@
 		}
 	}
 
+	if (startRow < 0) {
+		throw string("数字分隔未查找到起始行");
+	}
+
+	if (endRow < 0) {
+		throw string("数字分隔未查找到结束行");
+	}
+
 	binary = cv::Mat(binary, cv::Rect(0, startRow, cols, endRow - startRow + 1));
 	rows = binary.rows;
 	cols = binary.cols;
@@ -927,7 +935,15 @@
 
 cv::Mat ImgUtil::grayImage(cv::Mat src) {
 	cv::Mat grayImage;
-	cvtColor(src, grayImage, cv::COLOR_RGB2GRAY);
+	if (src.channels() == 3)
+	{
+		cvtColor(src, grayImage, cv::COLOR_RGB2GRAY);
+	}else if (src.channels() == 4) {
+		cvtColor(src, grayImage, cv::COLOR_RGBA2GRAY);
+	}
+	else if (src.channels() == 1) {
+		return src;
+	}
 	return grayImage;
 }
 

--
Gitblit v1.8.0