search

Qiuzhongyuan / pointpillars_pytorch_trt

Apache License 2.0
77 stars 24 forks source link

自定义plugin问题 #5

Open zhutongseu1994 opened 3 years ago

zhutongseu1994 commented 3 years ago

您好， 只在Trt.cpp中看到了initLibNvInferPlugins(&mLogger, "")初始化自定义plugin的函数，在其他地方test_onnx->CreateEngine（DeserializeEngine->BuildEngineWithOnnx->InitEngine）->Forward_mult 真的没看到自定义plugin

请问程序自定义了哪些plugin，是在工程的哪里呢？

谢谢您

chasingw commented 3 years ago

在 TensorRT7.2/plugin 里加了 3dnms scatter voxelgenerate 吧