diff --git a/paddlex/inference/models/object_detection/processors.py b/paddlex/inference/models/object_detection/processors.py index 5b473e0b3b..14a8e37754 100644 --- a/paddlex/inference/models/object_detection/processors.py +++ b/paddlex/inference/models/object_detection/processors.py @@ -623,7 +623,6 @@ def nms(boxes, iou_same=0.6, iou_diff=0.95): current = indices[0] current_box = boxes[current] current_class = current_box[0] - current_box[1] current_coords = current_box[2:] selected_boxes.append(current)