diff --git a/fb_detr/utils/non_max_supprs.py b/fb_detr/utils/non_max_supprs.py index 55811c7..f38a63e 100644 --- a/fb_detr/utils/non_max_supprs.py +++ b/fb_detr/utils/non_max_supprs.py @@ -12,7 +12,7 @@ def make_box(x1, y1, x2, y2): return dict(zip(keys, [x1, y1, x2, y2])) -def compute_intersection(a, b): # returns None if rectangles don't intersect +def compute_intersection(a, b): a = Rectangle(*a.values()) b = Rectangle(*b.values())