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

2
nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp
View File

@@ -25,7 +25,7 @@ upsampleLayer(int layerIdx, std::map<std::string, std::string>& block, nvinfer1:
std::string resizeLayerName = "upsample_" + std::to_string(layerIdx); std::string resizeLayerName = "upsample_" + std::to_string(layerIdx);
resize->setName(resizeLayerName.c_str()); resize->setName(resizeLayerName.c_str());
#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 4 #if NV_TENSORRT_MAJOR > 8 || (NV_TENSORRT_MAJOR == 8 && NV_TENSORRT_MINOR > 4)
resize->setResizeMode(nvinfer1::InterpolationMode::kNEAREST); resize->setResizeMode(nvinfer1::InterpolationMode::kNEAREST);
#else #else
resize->setResizeMode(nvinfer1::ResizeMode::kNEAREST); resize->setResizeMode(nvinfer1::ResizeMode::kNEAREST);

4
nvdsinfer_custom_impl_Yolo/yolo.cpp
View File

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