fix the region layer's output anchor size normalization error
authorzuoshaobo <zuoshaobo@science.com>
Thu, 14 Mar 2019 07:08:58 +0000 (03:08 -0400)
committerzuoshaobo <zuoshaobo@science.com>
Fri, 15 Mar 2019 12:41:11 +0000 (08:41 -0400)
modules/dnn/src/layers/region_layer.cpp

index cfe61c1..c33c1cb 100644 (file)
@@ -280,8 +280,8 @@ public:
 
                             dstData[box_index + 0] = (x + logistic_activate(srcData[box_index + 0])) / cols;
                             dstData[box_index + 1] = (y + logistic_activate(srcData[box_index + 1])) / rows;
-                            dstData[box_index + 2] = exp(srcData[box_index + 2]) * biasData[2 * a] / hNorm;
-                            dstData[box_index + 3] = exp(srcData[box_index + 3]) * biasData[2 * a + 1] / wNorm;
+                            dstData[box_index + 2] = exp(srcData[box_index + 2]) * biasData[2 * a] / wNorm;
+                            dstData[box_index + 3] = exp(srcData[box_index + 3]) * biasData[2 * a + 1] / hNorm;
 
                             int class_index = index_sample_offset + index * cell_size + 5;
                             for (int j = 0; j < classes; ++j) {