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

Fix build

Browse Source
This commit is contained in:
Marcos Luciano
2024-12-05 22:20:00 -03:00
parent e5d994e2d7
commit c11c3e4a78
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
nvdsinfer_custom_impl_Yolo/yolo.cpp
View File

@@ -77,7 +77,7 @@ Yolo::createEngine(nvinfer1::IBuilder* builder)
if (m_NetworkType == "onnx") {
#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
#if NV_TENSORRT_MAJOR > 8 || (NV_TENSORRT_MAJOR == 8 && NV_TENSORRT_MINOR > 0)
parser = nvonnxparser::createParser(*network, *builder->getLogger());
#else
parser = nvonnxparser::createParser(*network, logger);
@@ -193,7 +193,7 @@ Yolo::createEngine(nvinfer1::IBuilder* builder)
config->setProfilingVerbosity(nvinfer1::ProfilingVerbosity::kDETAILED);
#endif
#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
#if NV_TENSORRT_MAJOR > 8 || (NV_TENSORRT_MAJOR == 8 && NV_TENSORRT_MINOR > 0)
nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(*builder->getLogger());
#else
nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(logger);