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

Fix functions

Browse Source 
Fixed function for NVIDIA Jetson Nano
This commit is contained in:
Marcos Luciano
2020-12-21 23:34:49 -03:00
parent 6d541b58e7
commit cb99cf3254
4 changed files with 101 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
native/nvdsinfer_custom_impl_Yolo/yolo.cpp
View File

@@ -237,7 +237,7 @@ NvDsInferStatus Yolo::buildYoloNetwork(
= new YoloLayer(m_OutputTensors.at(outputTensorCount).numBBoxes,
m_OutputTensors.at(outputTensorCount).numClasses,
m_OutputTensors.at(outputTensorCount).gridSize,
'y', new_coords, scale_x_y, beta_nms,
1, new_coords, scale_x_y, beta_nms,
curYoloTensor.anchors,
m_OutputMasks);
assert(yoloPlugin != nullptr);
@@ -274,7 +274,7 @@ NvDsInferStatus Yolo::buildYoloNetwork(
= new YoloLayer(curRegionTensor.numBBoxes,
curRegionTensor.numClasses,
curRegionTensor.gridSize,
'r', 0, 1.0, 0,
0, 0, 1.0, 0,
curRegionTensor.anchors,
mask);
assert(regionPlugin != nullptr);