您好,我们自己定义的lenet模型在runtime_sim下的推理运行时报错:
请问这个问题该如何解决呢?
注释:这个问题我们在后处理源码文件中做了如下修改:
else if (model_name == "resnet18" || model_name == "googlenet" || model_name == "se_resnet_gray" || model_name == "mobilenetv1" || model_name == "mobilenetv2" || model_name == "resnet50_feature" || model_name == "lenet_gray" || model_name == "classification" || model_name.find("efficientnet") != std::string::npos) { return new ClassificationPostProcessModule(model_name);修改后如下else if (model_name == "resnet18" || model_name == "googlenet" || model_name=="lenet" || model_name == "se_resnet_gray" || model_name == "mobilenetv1" || model_name == "mobilenetv2" || model_name == "resnet50_feature" || model_name == "lenet_gray" || model_name == "classification" || model_name.find("efficientnet") != std::string::npos) { return new ClassificationPostProcessModule(model_name);虽然报错得以解决,但是图像输出的结果始终错误,错误如下: