From 76fd865537b5e04ee9fa64ba38ee2ba23affdce1 Mon Sep 17 00:00:00 2001
From: Jeffrey_Li <Jeffrey_Li@asus.com>
Date: Tue, 9 Sep 2025 15:56:36 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0yolo8=E7=AE=97=E5=AD=90?=
 =?UTF-8?q?=E5=86=85=E9=83=A8=E7=BA=BF=E7=A8=8B=E6=95=B0=E5=92=8C=E7=AE=97?=
 =?UTF-8?q?=E5=AD=90=E9=97=B4=E7=BA=BF=E7=A8=8B=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Cigarette/Alg/YOLOManager.h    | 3 ++-
 Cigarette/Alg/alg_jd_yolo8.cpp | 2 ++
 Cigarette/Alg/alg_jd_yolo8.h   | 1 +
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/Cigarette/Alg/YOLOManager.h b/Cigarette/Alg/YOLOManager.h
index 4ad43913..25c8c824 100644
--- a/Cigarette/Alg/YOLOManager.h
+++ b/Cigarette/Alg/YOLOManager.h
@@ -27,7 +27,8 @@ typedef struct _DL_INIT_PARAM
     int	keyPointsNum = 2;//Note:kpt number for pose
     bool cudaEnable = false;
     int logSeverityLevel = 3;
-    int intraOpNumThreads = 1;
+    int intraOpNumThreads = 8;
+    int interOpNumThreads = 8;
 } DL_INIT_PARAM;
 
 
diff --git a/Cigarette/Alg/alg_jd_yolo8.cpp b/Cigarette/Alg/alg_jd_yolo8.cpp
index 3e173db0..e8ca1c7c 100644
--- a/Cigarette/Alg/alg_jd_yolo8.cpp
+++ b/Cigarette/Alg/alg_jd_yolo8.cpp
@@ -105,6 +105,7 @@ bool YOLO_V8::InitParam(DL_INIT_PARAM& iParams)
     modelType = iParams.modelType;
     cudaEnable = iParams.cudaEnable;
     intraOpNumThreads = iParams.intraOpNumThreads;
+    interOpNumThreads = iParams.interOpNumThreads;
     logSeverityLevel = iParams.logSeverityLevel;
     classes = { "jd" };
     return true;
@@ -130,6 +131,7 @@ bool YOLO_V8::CreateSession(std::string model_path, std::string model_name) {
 
         sessionOption.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_ALL);
         sessionOption.SetIntraOpNumThreads(intraOpNumThreads);
+        sessionOption.SetInterOpNumThreads(interOpNumThreads);
         sessionOption.SetLogSeverityLevel(logSeverityLevel);
 
 #ifdef _WIN32
diff --git a/Cigarette/Alg/alg_jd_yolo8.h b/Cigarette/Alg/alg_jd_yolo8.h
index 967e4403..17a12fc7 100644
--- a/Cigarette/Alg/alg_jd_yolo8.h
+++ b/Cigarette/Alg/alg_jd_yolo8.h
@@ -56,6 +56,7 @@ private:
     //std::vector<const char*> outputNodeNames;
     int logSeverityLevel;
     int intraOpNumThreads;
+    int interOpNumThreads;
     MODEL_TYPE modelType;
     std::vector<int> imgSize;
     float rectConfidenceThreshold;