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

Fix logger error in DeepStream 6.0 / 6.0.1 + Change output classes format + Fixes

Browse Source
This commit is contained in:
Marcos Luciano
2023-06-08 13:47:43 -03:00
parent 9fd80c5248
commit 64fa573f72
23 changed files with 233 additions and 258 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nvdsinfer_custom_impl_Yolo/yolo.cpp
View File

@@ -76,7 +76,7 @@ Yolo::createEngine(nvinfer1::IBuilder* builder)
if (m_NetworkType == "onnx") {
#if NV_TENSORRT_MAJOR >= 8
#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
parser = nvonnxparser::createParser(*network, *builder->getLogger());
#else
parser = nvonnxparser::createParser(*network, logger);