修复yolo8没有class id的问题

main
Jeffrey_Li 4 days ago
parent 6ea27d23d7
commit 67ff7b2a52

@ -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(

Loading…
Cancel
Save