Update app.py

app.py

@@ -5,8 +5,120 @@ import PIL
 import torchvision.transforms as transforms
 
 ##Extras por si pudiera reconstruir la imagen en HF también
+import numpy as np
 import os
-import re
+import cv2
+
+def extract_subimages(image : np.ndarray, wwidth, wheight, overlap_fraction):
+    """
+    Extracts subimages of the input image using a moving window of size (wwidth, wheight)
+    with the specified overlap fraction. Returns a tuple (subimages, coords) where subimages
+    is a list of subimages and coords is a list of tuples (x, y) indicating the top left corner
+    coordinates of each subimage in the input image.
+    """
+    subimages = []
+    coords = []
+    height, width, channels = image.shape
+    if channels > 3:
+      image = image[:,:,0:3]
+      channels = 3
+    overlap = int(max(0, min(overlap_fraction, 1)) * min(wwidth, wheight))
+    y = 0
+    while y + wheight <= height:
+        x = 0
+        while x + wwidth <= width:
+            subimage = image[y:y+wheight, x:x+wwidth, :]
+            subimages.append(subimage)
+            coords.append((x, y))
+            x += wwidth - overlap
+        y += wheight - overlap
+    if y < height:
+        y = height - wheight
+        x = 0
+        while x + wwidth <= width:
+            subimage = image[y:y+wheight, x:x+wwidth, :]
+            subimages.append(subimage)
+            coords.append((x, y))
+            x += wwidth - overlap
+        if x < width:
+            x = width - wwidth
+            subimage = image[y:y+wheight, x:x+wwidth, :]
+            subimages.append(subimage)
+            coords.append((x, y))
+    if x < width:
+        x = width - wwidth
+        y = 0
+        while y + wheight <= height:
+            subimage = image[y:y+wheight, x:x+wwidth, :]
+            subimages.append(subimage)
+            coords.append((x, y))
+            y += wheight - overlap
+        if y < height:
+            y = height - wheight
+            subimage = image[y:y+wheight, x:x+wwidth, :]
+            subimages.append(subimage)
+            coords.append((x, y))
+    return subimages, coords
+
+# Si no hay archivos tif (labels) no se tratan, no hace falta considerarlo
+def generate_and_save_subimages(path, output_dir_images, output_dir_labels = None):
+  if output_dir_labels:
+    if not os.path.exists(output_dir_labels):
+        os.makedirs(output_dir_labels)
+
+  if not os.path.exists(output_dir_images):
+      os.makedirs(output_dir_images)
+
+  for filename in os.listdir(path):
+      if filename.endswith(".png") or filename.endswith(".tif"):
+          filepath = os.path.join(path, filename)
+          image = cv2.imread(filepath)
+          subimages, coords = extract_subimages(image, 400, 400, 0.66)
+          for i, subimage in enumerate(subimages):
+              if filename.endswith(".png"):
+                  output_filename = os.path.join(output_dir_images, f"{filename.rsplit('.', 1)[0]}_{coords[i][0]}_{coords[i][1]}.png")
+                  cv2.imwrite(output_filename, subimage)
+              else:
+                  if output_dir_labels:
+                    output_filename = os.path.join(output_dir_labels, f"{filename.rsplit('.', 1)[0]}_{coords[i][0]}_{coords[i][1]}.tif")
+                    cv2.imwrite(output_filename, subimage)
+
+def generate_and_save_subimages_nolabel(path, output_dir_images, olverlap=0.0, imagesformat="png", split_in_dirs=True):
+  for entry in os.scandir(path):
+    if entry.is_file() and entry.name.lower().endswith(imagesformat):
+      filepath = entry.path
+      gss_single(filepath, output_dir_images, olverlap, imagesformat, split_in_dirs)
+
+def gss_single(filepath, output_dir_images, olverlap=0.0, imagesformat="png", split_in_dirs=True):
+      image = cv2.imread(filepath)
+
+      if split_in_dirs:
+        dir_this_image = Path(output_dir_images)/filepath.rsplit('.', 1)[0]
+        os.makedirs(dir_this_image, exist_ok=True)
+      else:
+        os.makedirs(output_dir_images, exist_ok=True)
+
+      subimages, coords = extract_subimages(image, 400, 400, olverlap)
+      for i, subimage in enumerate(subimages):
+        if split_in_dirs:
+          output_filename = os.path.join(dir_this_image, f"{filepath.rsplit('.', 1)[0]}_{coords[i][0]}_{coords[i][1]}.png")
+        else:
+          output_filename = os.path.join(output_dir_images, f"{filepath.rsplit('.', 1)[0]}_{coords[i][0]}_{coords[i][1]}.png")
+        cv2.imwrite(output_filename, subimage)
+
+def split_windows_in_folders(input_images_folder, output_images_folder):
+  for filename in os.listdir(input_images_folder):
+    dir_this_image = Path(output_images_folder)/filename.rsplit('.', 1)[0]
+    os.makedirs(dir_this_image, exist_ok=True)
+    if filename.endswith(".png"):
+      print(str(dir_this_image))
+      filepath = os.path.join(path, filename)
+      image = cv2.imread(filepath)
+      subimages, coords = extract_subimages(image, 400, 400, 0)
+      for i, subimage in enumerate(subimages):
+          output_filename = os.path.join(dir_this_image, f"{filename.rsplit('.', 1)[0]}_{coords[i][0]}_{coords[i][1]}.png")
+          cv2.imwrite(output_filename, subimage)
+
 
 def subimages_from_directory(directorio):
   # Define el directorio a recorrer