IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] NMS. Fill the rest of first out blob. (#3251)

Browse Source
This commit is contained in:
Nikolay Shchegolev
2020-11-22 20:06:15 +03:00
committed by GitHub
parent ec8527f21e
commit 1da3539bc1
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
inference-engine/src/mkldnn_plugin/nodes/non_max_suppression.cpp
View File

@@ -379,13 +379,15 @@ public:
});
}
size_t validOutputs = std::min(filtBoxes.size(), static_cast<size_t>(outputs[NMS_SELECTEDINDICES]->getTensorDesc().getDims()[0]));
const size_t selectedBoxesNum = outputs[NMS_SELECTEDINDICES]->getTensorDesc().getDims()[0];
const size_t validOutputs = std::min(filtBoxes.size(), selectedBoxesNum);
int selectedIndicesStride = outputs[NMS_SELECTEDINDICES]->getTensorDesc().getBlockingDesc().getStrides()[0];
int *selectedIndicesPtr = selected_indices;
float *selectedScoresPtr = selected_scores;
for (size_t idx = 0; idx < validOutputs; idx++) {
size_t idx = 0lu;
for (; idx < validOutputs; idx++) {
selectedIndicesPtr[0] = filtBoxes[idx].batch_index;
selectedIndicesPtr[1] = filtBoxes[idx].class_index;
selectedIndicesPtr[2] = filtBoxes[idx].box_index;
@@ -397,6 +399,7 @@ public:
selectedScoresPtr += selectedIndicesStride;
}
}
std::fill(selectedIndicesPtr, selectedIndicesPtr + (selectedBoxesNum - idx) * selectedIndicesStride, -1);
if (outputs.size() > NMS_VALIDOUTPUTS)
*valid_outputs = static_cast<int>(validOutputs);