[IE CLDNN] Fixed a region_yolo bug (#51761) (openvinotoolkit#5826)

Signed-off-by: Kim, SungEun <[email protected]>
yekruglov · Jun 7, 2021 · edf7b2f · edf7b2f
1 parent f05e6ef
commit edf7b2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/region_yolo_gpu_ref.cl b/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/region_yolo_gpu_ref.cl
@@ -10,7 +10,7 @@ inline INPUT0_TYPE FUNC(logistic_activate)(INPUT0_TYPE x) {
 
 inline int FUNC(output_index)(int batch, int region_num, int x, int y, int xy, int feature_offset) {
 #if DO_SOFTMAX
-    return OUTPUT_GET_INDEX(batch, feature_offset * INPUT0_SIZE_X * INPUT0_SIZE_Y + xy, 1, 1);
+    return OUTPUT_GET_INDEX(batch, feature_offset * INPUT0_SIZE_X * INPUT0_SIZE_Y + xy, 0, 0);
 #else
     return OUTPUT_GET_INDEX(batch, feature_offset, y, x);
 #endif