Update square_crop.py

square_crop.py

@@ -20,15 +20,15 @@ def load_coco(json_path):
     return id2fname, coco["annotations"]
 
 
-def square_from_bbox(x, y, w, h, img_w, img_h):
+def square_from_bbox(x1, y1, x2, y2, img_w, img_h):
     """
     Compute (left, top, side) of the smallest square fully containing the bbox.
     The square is centred on the bbox; if it overflows the image, it is shifted
     (but not resized) so it lies inside the image.  Returns the final crop box
     (left, top, side).
     """
-    side = max(w, h)
-    cx, cy = x + w / 2.0, y + h / 2.0
+    side = max(x2-x1, y2-y1)
+    cx, cy = (x1+x2) / 2.0, (y1+y2) / 2.0
     left = int(round(cx - side / 2.0))
     top = int(round(cy - side / 2.0))
 
@@ -41,8 +41,8 @@ def square_from_bbox(x, y, w, h, img_w, img_h):
 def crop_annotation(img_path, ann, out_dir, pad_color=0):
     with Image.open(img_path) as img:
         img_w, img_h = img.size
-        x, y, w, h = ann["bbox"]          # COCO bbox = [x, y, width, height]
-        left, top, side = square_from_bbox(x, y, w, h, img_w, img_h)
+        x1, y1, x2, y2 = ann["bbox"]
+        left, top, side = square_from_bbox(x1, y1, x2, y2, img_w, img_h)
 
         # Perform crop (may be smaller than 'side' at edges)
         crop = img.crop((left, top, left + side, top + side))