From 67ff7b2a52200893016d19292f10deda91494464 Mon Sep 17 00:00:00 2001
From: Jeffrey_Li <Jeffrey_Li@asus.com>
Date: Tue, 9 Sep 2025 14:09:18 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dyolo8=E6=B2=A1=E6=9C=89class?=
 =?UTF-8?q?=20id=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Cigarette/Alg/alg_jd_yolo8.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Cigarette/Alg/alg_jd_yolo8.cpp b/Cigarette/Alg/alg_jd_yolo8.cpp
index 912b3887..3e173db0 100644
--- a/Cigarette/Alg/alg_jd_yolo8.cpp
+++ b/Cigarette/Alg/alg_jd_yolo8.cpp
@@ -106,6 +106,7 @@ bool YOLO_V8::InitParam(DL_INIT_PARAM& iParams)
     cudaEnable = iParams.cudaEnable;
     intraOpNumThreads = iParams.intraOpNumThreads;
     logSeverityLevel = iParams.logSeverityLevel;
+    classes = { "jd" };
     return true;
 }
 
@@ -224,7 +225,7 @@ char* YOLO_V8::RunSession(cv::Mat& in, cv::Mat& out, std::vector<std::pair<int,
 
         float confidence = floor(100 * re.confidence) / 100;
         std::cout << std::fixed << std::setprecision(2);
-        std::string label = std::string("jd") + std::string(" ") +
+        std::string label = classes[re.classId] + std::string(" ") +
             std::to_string(confidence).substr(0, std::to_string(confidence).size() - 4);
 
         cv::rectangle(