IndexError Traceback (most recent call last)
in ()
6 STRIDES = np.array(STRIDES)
7
----> 8 pred_bbox = postprocess_bbbox(detections, ANCHORS, STRIDES, XYSCALE)
9 bboxes = postprocess_boxes(pred_bbox, original_image_size, input_size, 0.25)
10 bboxes = nms(bboxes, 0.213, method='nms')
in postprocess_bbbox(pred_bbox, ANCHORS, STRIDES, XYSCALE)
14 conv_shape = pred.shape
15 output_size = conv_shape[1]
---> 16 conv_raw_dxdy = pred[:, :, :, :, 0:2]
17 conv_raw_dwdh = pred[:, :, :, :, 2:4]
18 xy_grid = np.meshgrid(np.arange(output_size), np.arange(output_size))
IndexError: too many indices for array
models copied to clipboard
onnx
