From 530ba7b3bf4608c90da0b2a31c5b9df92523b39f Mon Sep 17 00:00:00 2001 From: liusiyang Date: Tue, 21 Jul 2026 10:56:32 +0800 Subject: [PATCH] =?UTF-8?q?update=20=E6=95=B4=E7=90=86=E6=A0=87=E7=AD=BE?= =?UTF-8?q?=E5=AD=98=E5=9B=BE=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- AlgorithmModule/src/ImgCheckAnalysisy.cpp | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/AlgorithmModule/src/ImgCheckAnalysisy.cpp b/AlgorithmModule/src/ImgCheckAnalysisy.cpp index 49343f9..e14449b 100644 --- a/AlgorithmModule/src/ImgCheckAnalysisy.cpp +++ b/AlgorithmModule/src/ImgCheckAnalysisy.cpp @@ -2189,6 +2189,9 @@ int ImgCheckAnalysisy::AI_Detect_BQ() m_pdetlog->AddCheckstr(PrintLevel_1, strBaseLog, "============= Start, BQ num = %d", bqNum); long ts = CheckUtil::getcurTime(); + std::string BQ_save_path = "/home/aidlux/BOE/FOG/BQ_AI_Det/"+ m_pImageAllResult->productBaseResult->strproductName + "/"; + CheckUtil::CreateDir(BQ_save_path); + // 初始化BQ AI检测结果 m_pImageAllResult->bq_DetAIResult = std::make_shared(); std::shared_ptr pDetAIResult = m_pImageAllResult->bq_DetAIResult; @@ -2285,7 +2288,7 @@ int ImgCheckAnalysisy::AI_Detect_BQ() { cv::rectangle(debugImg, BQ_SmallRoiList.at(i), cv::Scalar(0, 0, 255), 2); } - cv::imwrite(m_strCurDetCamChannel + "_BQ_original_with_rois.png", debugImg); + cv::imwrite(BQ_save_path + m_strCurDetCamChannel + "_BQ_original_with_rois.png", debugImg); } int submitted = 0; @@ -2307,7 +2310,7 @@ int ImgCheckAnalysisy::AI_Detect_BQ() // 调试存图：保存每个小块 if (m_pImageAllResult->cameraBaseResult->pBQ_Result->bsaveAll) { - cv::imwrite(m_strCurDetCamChannel + "_BQ_crop_" + std::to_string(submitted) + ".png", task->input); + cv::imwrite(BQ_save_path +m_strCurDetCamChannel + "_BQ_crop_" + std::to_string(submitted) + ".png", task->input); } runner->SubmitTask(task); @@ -2324,7 +2327,7 @@ int ImgCheckAnalysisy::AI_Detect_BQ() // 调试存图：保存模型原始输出mask if (m_pImageAllResult->cameraBaseResult->pBQ_Result->bsaveAll) { - cv::imwrite(m_strCurDetCamChannel + "_BQ_outmask_" + std::to_string(result->id) + ".png", outimg); + cv::imwrite(BQ_save_path + m_strCurDetCamChannel + "_BQ_outmask_" + std::to_string(result->id) + ".png", outimg); } cv::Mat AIresult; @@ -2341,9 +2344,9 @@ int ImgCheckAnalysisy::AI_Detect_BQ() AIresult.copyTo(AIresultMask(srcsize_roi), AIresult); // 调试存图：保存resize后的mask - if (DetImgInfo_shareP->bDebugsaveImg && bResizeToSrc) + if (m_pImageAllResult->cameraBaseResult->pBQ_Result->bsaveAll && bResizeToSrc) { - cv::imwrite(m_strCurDetCamChannel + "_BQ_mask_resized_" + std::to_string(result->id) + ".png", AIresult); + cv::imwrite(BQ_save_path + m_strCurDetCamChannel + "_BQ_mask_resized_" + std::to_string(result->id) + ".png", AIresult); } m_pdetlog->AddCheckstr(PrintLevel_2, strBaseLog, "BQ[%d] AI done, roi=[%d,%d,%d,%d]", @@ -2366,9 +2369,9 @@ int ImgCheckAnalysisy::AI_Detect_BQ() m_pdetlog->AddCheckstr(PrintLevel_1, strBaseLog, "============= End; AI Run Time: sum %ld pre %ld Run %ld mean One Small Img %f", te - ts, t2 - ts, te - t2, mean_AI); // 调试存图 - if (DetImgInfo_shareP->bDebugsaveImg) + if (m_pImageAllResult->cameraBaseResult->pBQ_Result->bsaveAll) { - cv::imwrite(m_strCurDetCamChannel + "_AI_BQ_mask.png", pDetAIResult->AI_MaskImg); + cv::imwrite(BQ_save_path + m_strCurDetCamChannel + "_AI_BQ_mask.png", pDetAIResult->AI_MaskImg); } return 0;