diff --git a/AIEngineModule/src/AIModel_Impl.cpp b/AIEngineModule/src/AIModel_Impl.cpp index faeb525..476c2db 100644 --- a/AIEngineModule/src/AIModel_Impl.cpp +++ b/AIEngineModule/src/AIModel_Impl.cpp @@ -600,8 +600,8 @@ int AIModel_Impl::F2softmaxId(float *data, int class_num, float *fmaxScore) std::vector output(cls_num); for (int i = 0; i < output.size(); i++) { - float outi = (float)exp(data[i]) / (float)total; - output[i] = outi; + // float outi = (float)exp(data[i]) / (float)total; + output[i] = data[i]; // 不做softmax处理(模型已经做了) // output.push_back(outi); // std::cout<<"output["<cls_score); + std::string strclassName = AI_CLass_QX_NAME_Names[cls_num]; + m_pdetlog->AddCheckstr(PrintLevel_1, "AI Class num", "%d = %s %f\n", cls_num, strclassName.c_str(), result->cls_score); // 分类存图 开启。 if (m_pbaseCheckFunction->saveImg.bSaveClsImg) @@ -2186,7 +2189,7 @@ int ImgCheckAnalysisy::AIClassTypeToConfigType(int nAIQXType) { // 无对应缺陷类型时返回 CONFIG_QX_NAME_count(缺陷列表已移除 other) - int resultError_type = CONFIG_QX_NAME_count; + int resultError_type = AI_CLass_QX_NAME_Panel; switch (nAIQXType) { case AI_CLass_QX_NAME_PIN_PoSun: @@ -2210,6 +2213,9 @@ int ImgCheckAnalysisy::AIClassTypeToConfigType(int nAIQXType) case AI_CLass_QX_NAME_C_BianXing: resultError_type = CONFIG_QX_NAME_C_BianXing; break; + case AI_CLass_QX_NAME_Panel: + resultError_type = CONFIG_QX_NAME_Panel; + break; default: break; }