diff --git a/deploy/pphuman/pipe_utils.py b/deploy/pphuman/pipe_utils.py index 43904e657c1a4921adb6656ecb25682aa7d0f984..4c7601a5308ecdcdaf6478480012b7168580dbd0 100644 --- a/deploy/pphuman/pipe_utils.py +++ b/deploy/pphuman/pipe_utils.py @@ -297,6 +297,8 @@ def crop_image_with_det(batch_input, det_res, thresh=0.3): crop_res = [] for b_id, input in enumerate(batch_input): boxes_num_i = boxes_num[b_id] + if boxes_num_i == 0: + continue boxes_i = boxes[start_idx:start_idx + boxes_num_i, :] score_i = score[start_idx:start_idx + boxes_num_i] res = []