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

Merge pull request #36 from pieris98/master

Browse Source 
Comment leftover assertions of square dims
This commit is contained in:
Marcos Luciano
2021-02-19 15:40:36 -03:00
committed by GitHub
parent 3f3ccbeb73 8d5847c658
commit c69f50a75e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
non_square/nvdsinfer_custom_impl_Yolo/yolo.cpp
View File

@@ -212,7 +212,7 @@ NvDsInferStatus Yolo::buildYoloNetwork(
else if (m_ConfigBlocks.at(i).at("type") == "yolo") { else if (m_ConfigBlocks.at(i).at("type") == "yolo") {
nvinfer1::Dims prevTensorDims = previous->getDimensions(); nvinfer1::Dims prevTensorDims = previous->getDimensions();
assert(prevTensorDims.d[1] == prevTensorDims.d[2]); //assert(prevTensorDims.d[1] == prevTensorDims.d[2]);
TensorInfo& curYoloTensor = m_OutputTensors.at(outputTensorCount); TensorInfo& curYoloTensor = m_OutputTensors.at(outputTensorCount);
curYoloTensor.gridSizeY = prevTensorDims.d[1]; curYoloTensor.gridSizeY = prevTensorDims.d[1];
curYoloTensor.gridSizeX = prevTensorDims.d[2]; curYoloTensor.gridSizeX = prevTensorDims.d[2];
@@ -262,7 +262,7 @@ NvDsInferStatus Yolo::buildYoloNetwork(
//YOLOv2 support //YOLOv2 support
else if (m_ConfigBlocks.at(i).at("type") == "region") { else if (m_ConfigBlocks.at(i).at("type") == "region") {
nvinfer1::Dims prevTensorDims = previous->getDimensions(); nvinfer1::Dims prevTensorDims = previous->getDimensions();
assert(prevTensorDims.d[1] == prevTensorDims.d[2]); //assert(prevTensorDims.d[1] == prevTensorDims.d[2]);
TensorInfo& curRegionTensor = m_OutputTensors.at(outputTensorCount); TensorInfo& curRegionTensor = m_OutputTensors.at(outputTensorCount);
curRegionTensor.gridSizeY = prevTensorDims.d[1]; curRegionTensor.gridSizeY = prevTensorDims.d[1];
curRegionTensor.gridSizeX = prevTensorDims.d[2]; curRegionTensor.gridSizeX = prevTensorDims.d[2];
@@ -466,4 +466,4 @@ void Yolo::destroyNetworkUtils() {
free(const_cast<void*>(m_TrtWeights[i].values)); free(const_cast<void*>(m_TrtWeights[i].values));
} }
m_TrtWeights.clear(); m_TrtWeights.clear();
} }