Update square_crop.py

square_crop.py

@@ -47,6 +47,8 @@ def crop_annotation(img_path, ann, out_dir, pad_color=0):
         # Perform crop (may be smaller than 'side' at edges)
         crop = img.crop((left, top, left + side, top + side))
 
+        print(f"final image {crop.size} from original image {img.size}")
+
         # If we lost pixels at the edge, pad back to full square
         if crop.size != (side, side):
             delta_w = side - crop.size[0]
@@ -78,4 +80,5 @@ def run_square_crop(input_dir, coco_json_path, cropped_dir):
             # except Exception as e:
             #     print(f"Error on {img_path}.")
 
+    print(os.listdir(cropped_dir))
     return os.listdir(cropped_dir)