added pad_to_square to io_utils.py

io_utils.py

@@ -2,13 +2,37 @@
 # This files contains OSAIL utils to read and write files.
 ################################################################################
 
-from .data import pad_to_square
 import copy
 import monai as mn
 import numpy as np
 import os
 import skimage
 
+################################################################################
+# -F: pad_to_square
+
+def pad_to_square(image):
+    """A function to pad an image to square shape with zero pixels.
+
+    Args:
+        image (np.ndarray): the input image array.
+
+    Returns:
+        np.ndarray: the padded image array.
+    """
+    height, width = image.shape
+    if height < width: 
+        padded_image = np.zeros((width, width))
+        delta = (width - height) // 2
+        padded_image[delta:height+delta, :] = image
+        image = padded_image
+    elif height > width:
+        padded_image = np.zeros((height, height))
+        delta = (height - width) // 2
+        padded_image[:, delta:width+delta] = image
+        image = padded_image
+    return image
+
 ################################################################################
 # -F: load_image