thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix DeepStream 5 and 6

Browse Source
This commit is contained in:
Marcos Luciano
2024-11-13 10:10:12 -03:00
parent b451b036b2
commit 1b96a01ec3
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
nvdsinfer_custom_impl_Yolo/yolo.cpp
View File

@@ -193,7 +193,12 @@ Yolo::createEngine(nvinfer1::IBuilder* builder)
config->setProfilingVerbosity(nvinfer1::ProfilingVerbosity::kDETAILED); config->setProfilingVerbosity(nvinfer1::ProfilingVerbosity::kDETAILED);
#endif #endif
#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(*builder->getLogger()); nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(*builder->getLogger());
#else
nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(logger);
#endif
assert(runtime); assert(runtime);
nvinfer1::IHostMemory* serializedEngine = builder->buildSerializedNetwork(*network, *config); nvinfer1::IHostMemory* serializedEngine = builder->buildSerializedNetwork(*network, *config);