diff --git a/python/caffe/layers/detectnet/clustering.py b/python/caffe/layers/detectnet/clustering.py index 9547db7d880..77bec07387b 100644 --- a/python/caffe/layers/detectnet/clustering.py +++ b/python/caffe/layers/detectnet/clustering.py @@ -150,7 +150,8 @@ def gridbox_to_boxes(net_cvg, net_boxes, self): cell_width = im_sz_x / grid_sz_x cell_height = im_sz_y / grid_sz_y - cvg_val = net_cvg[0, 0:grid_sz_y, 0:grid_sz_x] + #cvg_val = net_cvg[0, 0:grid_sz_y, 0:grid_sz_x] + cvg_val = net_cvg[0][0:net_cvg.shape[1],0:net_cvg.shape[2]] if (self.is_groundtruth): mask = (cvg_val > 0)