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

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

Browse Source 
Signed-off-by: Kim, SungEun <sungeun.kim@intel.com>
This commit is contained in:
Sungeun Kim 2021-05-27 02:21:42 +09:00 committed by GitHub
parent 5adf2ee80e
commit 163d0b27b0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/region_yolo_gpu_ref.cl vendored
View File

@ -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