liusiyang
/
BOE_FOG_DETECT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update 整理标签存图路径

Browse Source
dev_lsy
liusiyang 3 weeks ago
parent a5bfe40bee
commit 530ba7b3bf
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File

17
AlgorithmModule/src/ImgCheckAnalysisy.cpp
Unescape View File

@ -2189,6 +2189,9 @@ int ImgCheckAnalysisy::AI_Detect_BQ()
m_pdetlog->AddCheckstr(PrintLevel_1, strBaseLog, "============= Start, BQ num = %d", bqNum); m_pdetlog->AddCheckstr(PrintLevel_1, strBaseLog, "============= Start, BQ num = %d", bqNum);
long ts = CheckUtil::getcurTime(); 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检测结果 // 初始化BQ AI检测结果
m_pImageAllResult->bq_DetAIResult = std::make_shared<ImageAllResult::Image_AI_Det_Result>(); m_pImageAllResult->bq_DetAIResult = std::make_shared<ImageAllResult::Image_AI_Det_Result>();
std::shared_ptr<ImageAllResult::Image_AI_Det_Result> pDetAIResult = m_pImageAllResult->bq_DetAIResult; std::shared_ptr<ImageAllResult::Image_AI_Det_Result> 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::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; int submitted = 0;
@ -2307,7 +2310,7 @@ int ImgCheckAnalysisy::AI_Detect_BQ()
// 调试存图：保存每个小块 // 调试存图：保存每个小块
if (m_pImageAllResult->cameraBaseResult->pBQ_Result->bsaveAll) 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); runner->SubmitTask(task);
@ -2324,7 +2327,7 @@ int ImgCheckAnalysisy::AI_Detect_BQ()
// 调试存图：保存模型原始输出mask // 调试存图：保存模型原始输出mask
if (m_pImageAllResult->cameraBaseResult->pBQ_Result->bsaveAll) 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; cv::Mat AIresult;
@ -2341,9 +2344,9 @@ int ImgCheckAnalysisy::AI_Detect_BQ()
AIresult.copyTo(AIresultMask(srcsize_roi), AIresult); AIresult.copyTo(AIresultMask(srcsize_roi), AIresult);
// 调试存图：保存resize后的mask // 调试存图：保存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]", 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); 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; return 0;

Write Preview
Loading…
Cancel
Save