update 分类不做softmax

dev_lsy
liusiyang 1 week ago
parent 4e95e93a91
commit e8fa455ae8

@ -599,8 +599,8 @@ int AIModel_Impl::F2softmaxId(float *data, int class_num, float *fmaxScore)
std::vector<float> 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["<<i<<"]="<<output[i]<<std::endl;
}

Loading…
Cancel
Save