Update region_layer.cpp

Fix objectness (dstData[index + 4]) is not assigned if new_coords == 1.
2 years ago · c8561eae2d
parent 2c81cb2f76
commit c8561eae2d
1 changed files with 1 additions and 0 deletions
--- a/modules/dnn/src/layers/region_layer.cpp
+++ b/modules/dnn/src/layers/region_layer.cpp
@ -317,6 +317,7 @@ public:
                                dstData[box_index + 1] = (y + y_tmp) / rows;
                                dstData[box_index + 2] = (srcData[box_index + 2]) * (srcData[box_index + 2]) * 4 * biasData[2 * a] / wNorm;
                                dstData[box_index + 3] = (srcData[box_index + 3]) * (srcData[box_index + 3]) * 4 * biasData[2 * a + 1] / hNorm;
+                                dstData[box_index + 4] = srcData[p_index];

                                scale = srcData[p_index];
                                if (classfix == -1 && scale < thresh)