部署PointPillars出错

跑autoware的PointPillars 遇到如下错误

ERROR: ModelImporter.cpp:472 In function importModel:
[4] Assertion failed: !_importer_ctx.network()->hasImplicitBatchDimension() && "This version of the ONNX parser only supports TensorRT INetworkDefinitions with an explicit batch dimension. Please ensure the network was created using the EXPLICIT_BATCH NetworkDefinitionCreationFlag."
ERROR: failed to parse onnx file

在这里找到了解决办法

https://www.cnblogs.com/cyssmile/p/15399109.html

将nvinfer1::INetworkDefinition* network = builder->createNetworkV2(0);的0改成1即可;
或者将nvinfer1::INetworkDefinition *network = builder->createNetwork()改成nvinfer1::INetworkDefinition *network = builder->createNetworkV2(1)