张洋 2 дней назад
Родитель
Сommit
4990ad59ae

BIN
res/ai/best.onnx


BIN
res/ai/yolo26s-cls.onnx


BIN
res/ai/yolo26n-cls.onnx


BIN
res/ai/yolo26n.onnx


+ 1 - 1
zhipuzi_pos_windows/ai/YoloFeatureExtractor.cpp

@@ -11,7 +11,7 @@
 #include "../tool/debuglog.h"
 #include "../tool/debuglog.h"
 
 
 YoloFeatureExtractor::YoloFeatureExtractor(const std::string& modelPath, const std::string& classesPath)
 YoloFeatureExtractor::YoloFeatureExtractor(const std::string& modelPath, const std::string& classesPath)
-	: inputWidth(640), inputHeight(640)
+	: inputWidth(448), inputHeight(448)
 {
 {
 	net = cv::dnn::readNetFromONNX(modelPath);
 	net = cv::dnn::readNetFromONNX(modelPath);
 	loadClassNames(classesPath);
 	loadClassNames(classesPath);

+ 1 - 1
zhipuzi_pos_windows/ai/test.cpp

@@ -25,7 +25,7 @@ int AITest()
 		//用于测试的图片目录
 		//用于测试的图片目录
 		std::string galleryDir = (mainDir.parent_path().parent_path().parent_path().parent_path() /"res"/"images").string();       // 图库目录路径
 		std::string galleryDir = (mainDir.parent_path().parent_path().parent_path().parent_path() /"res"/"images").string();       // 图库目录路径
 
 
-		std::string modelPath = sMainDir + "/ai/best.onnx";           // YOLO2026模型路径
+		std::string modelPath = sMainDir + "/ai/best_448.onnx";           // YOLO2026模型路径
 		std::string classesPath = sMainDir + "/ai/cls.names";             // 类别文件路径
 		std::string classesPath = sMainDir + "/ai/cls.names";             // 类别文件路径
 		std::string searchImagePath = sMainDir + "/3.jpg"; // 搜索图片路径
 		std::string searchImagePath = sMainDir + "/3.jpg"; // 搜索图片路径
 
 

+ 1 - 1
zhipuzi_pos_windows/zhipuzi_pos_windows.cpp

@@ -26,7 +26,7 @@ int APIENTRY wWinMain(_In_ HINSTANCE hInstance,
 	CLewaimaiLog log;
 	CLewaimaiLog log;
 	log.Init();
 	log.Init();
 
 
-	//AITest();
+	AITest();
 
 
 	//初始化所有配置,启动所有异步worker
 	//初始化所有配置,启动所有异步worker
 	CAppEnv app;
 	CAppEnv app;