mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 06:26:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dnn: fix variance setting bug for PriorBoxLayer

Browse Source 
- The size of second channel should be size[2] of output tensor,
- The Scalar should be {variance[0], variance[0], variance[0], variance[0]}
  for _variance.size() == 1 case.

Signed-off-by: Wu Zhiwen <zhiwen.wu@intel.com>
This commit is contained in:
Wu Zhiwen 2018-08-29 11:47:13 +08:00
parent 087e9308ec
commit ca51bbb7ff
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/dnn/src/layers/prior_box_layer.cpp
View File

@ -453,8 +453,8 @@ public:
outputPtr = outputs[0].ptr<float>(0, 1); outputPtr = outputs[0].ptr<float>(0, 1);
if(_variance.size() == 1) if(_variance.size() == 1)
{ {
Mat secondChannel(outputs[0].size[2], outputs[0].size[3], CV_32F, outputPtr); Mat secondChannel(1, outputs[0].size[2], CV_32F, outputPtr);
secondChannel.setTo(Scalar(_variance[0])); secondChannel.setTo(Scalar::all(_variance[0]));
} }
else else
{ {