Create crop_breast_region.py

crop_breast_region.py

@@ -0,0 +1,28 @@
+import os
+from PIL import Image
+path = "/home/user/app/image_folder"
+crop_dst = '/home/user/app/cropped_images'
+for img_name in os.listdir(path):
+  image = Image.open(os.path.join(path,img_name))
+  # Get the original image dimensions
+  image_width, image_height = image.size
+  with open('runs/detect/exp/labels/'+img_name[:-4]+'.txt') as f:
+    for line in f:
+      b,x,y,width,height = line.split(' ')
+      x,y,width,height = float(x),float(y),float(width),float(height)
+      # Convert YOLO coordinates to absolute coordinates
+      abs_x = x * image_width
+      abs_y = y * image_height
+      abs_width = width * image_width
+      abs_height = height * image_height
+
+      # Calculate the region coordinates
+      xmin_crop = int(abs_x - abs_width/2)
+      ymin_crop = int(abs_y - abs_height/2)
+      xmax_crop = int(abs_x + abs_width/2)
+      ymax_crop = int(abs_y + abs_height/2)
+
+      # Extract the region using absolute coordinates
+      region = image.crop((xmin_crop, ymin_crop, xmax_crop, ymax_crop))
+      width_crop,height_crop = region.size
+      region.save(os.path.join(crop_dst,img_name))