diff --git a/net/utilities.py b/net/utilities.py index 6c4bc1f..109ad25 100755 --- a/net/utilities.py +++ b/net/utilities.py @@ -542,10 +542,7 @@ def get_detections_after_soft_non_maximum_suppression(detections, sigma, score_t max_x = np.minimum(detections[0, 2], detections[1:, 2]) max_y = np.minimum(detections[0, 3], detections[1:, 3]) - overlap_width = np.maximum(max_x - min_x + 1, 0.0) - overlap_height = np.maximum(max_y - min_y + 1, 0.0) - - intersection_area = overlap_width * overlap_height + intersection_area = np.maximum(max_x - min_x + 1, 0.0) * np.maximum(max_y - min_y + 1, 0.0) intersection_over_union = \ intersection_area / (detections[0, areas_index] + detections[1:, areas_index] - intersection_area)