sair-lab / AirSLAM

🚀 AirVO upgrades to AirSLAM 🚀
GNU General Public License v3.0
714 stars 102 forks source link

construct_network fail #115

Open YZH-bot opened 1 month ago

YZH-bot commented 1 month ago

@xukuanHIT 您好,请问这里parse模型返回空,导致parse失败有可能是什么原因呢?

bool SuperGlue::construct_network(TensorRTUniquePtr<nvinfer1::IBuilder> &builder,
                                  TensorRTUniquePtr<nvinfer1::INetworkDefinition> &network,
                                  TensorRTUniquePtr<nvinfer1::IBuilderConfig> &config,
                                  TensorRTUniquePtr<nvonnxparser::IParser> &parser) const
{
    auto parsed = parser->parseFromFile(superglue_config_.onnx_file.c_str(),
                                        static_cast<int>(gLogger.getReportableSeverity()));
    if (!parsed)
    {
        return false;
    }
    config->setMaxWorkspaceSize(512_MiB);
    config->setFlag(BuilderFlag::kFP16);
    enableDLA(builder.get(), config.get(), superglue_config_.dla_core);
    return true;
}
xukuanHIT commented 1 month ago

@YZH-bot 你好,单靠提供的信息难以定位问题。不过你可以检查下文件路径是否正确,环境配置是否有问题