diff --git a/AIEngineModule/src/AIModel_Impl.cpp b/AIEngineModule/src/AIModel_Impl.cpp index 4c07740..a9b03d2 100644 --- a/AIEngineModule/src/AIModel_Impl.cpp +++ b/AIEngineModule/src/AIModel_Impl.cpp @@ -599,8 +599,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["<