From 67a5e3d825fde17b7b00906ce42b0bd8cafebc4e Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期四, 18 八月 2022 18:22:14 +0800
Subject: [PATCH] '完善'

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

diff --git a/ConsoleApplication/ImgUtil.cpp b/ConsoleApplication/ImgUtil.cpp
index 0a9d58a..fca91be 100644
--- a/ConsoleApplication/ImgUtil.cpp
+++ b/ConsoleApplication/ImgUtil.cpp
@@ -1,4 +1,5 @@
 #include "ImgUtil.h"
+#include <io.h>
 
 NumberData ImgUtil::NUMS_LEVEL2[10];
 NumberData ImgUtil::NUMS_GP_CODE[10];
@@ -423,6 +424,11 @@
 	//LEVE2数字初始化
 	for (int i = 0;i < 10;i++) {
 		std::string sdir = "C:\\Users\\Administrator\\Desktop\\ocr\\number\\";
+		if (_access(sdir.c_str(), 0)) {
+			sdir = "sample\\number\\";
+		}
+
+
 		std::string path = sdir.append(std::to_string(i)).append(".jpg");
 		NumberData data = NumberData();
 		data.data = formatNumLevel2(cv::imread(path.c_str(), cv::IMREAD_GRAYSCALE));
@@ -445,6 +451,10 @@
 
 	for (int i = 0;i < 10;i++) {
 		std::string sdir = "C:\\Users\\Administrator\\Desktop\\ocr\\number_2\\";
+		if (_access(sdir.c_str(), 0)) {
+			sdir = "sample\\number_2\\";
+		}
+
 		std::string path = sdir.append(std::to_string(i)).append(".jpg");
 		NumberData data = NumberData();
 		data.data = formatNumGPCode(cv::imread(path.c_str(), cv::IMREAD_GRAYSCALE));

--
Gitblit v1.8.0