replaced load_image

app.py

@@ -7,10 +7,10 @@ import gradio as gr
 import jax
 import numpy as np
 import spaces
-import zea
 from PIL import Image
 
 from main import Config, init, run
+from utils import load_image
 
 CONFIG_PATH = "configs/semantic_dps.yaml"
 SLIDER_CONFIG_PATH = "configs/slider_params.yaml"
@@ -127,7 +127,7 @@ example_image_paths = [
     for f in os.listdir(ASSETS_DIR)
     if f.lower().endswith(".png")
 ]
-example_images = [zea.io_lib.load_image(p) for p in example_image_paths]
+example_images = [load_image(p) for p in example_image_paths]
 examples = [[img] for img in example_images]
 
 

eval.py

@@ -9,10 +9,10 @@ from PIL import Image
 from scipy.ndimage import binary_erosion, distance_transform_edt
 from scipy.stats import ks_2samp
 from zea import log
-from zea.io_lib import load_image
 
 import fid_score
 from plots import plot_metrics
+from utils import load_image
 
 
 def calculate_fid_score(denoised_image_dirs, ground_truth_dir):

main.py

@@ -28,6 +28,7 @@ from plots import create_animation, plot_batch_with_named_masks, plot_dehazed_re
 from utils import (
     apply_bottom_preservation,
     extract_skeleton,
+    load_image,
     postprocess,
     preprocess,
     smooth_L1,
@@ -374,7 +375,7 @@ def main(
 
     images = []
     for path in paths:
-        image = zea.io_lib.load_image(path)
+        image = load_image(path)
         images.append(image)
     images = ops.stack(images, axis=0)
 

sweeper.py

@@ -23,6 +23,7 @@ from zea import init_device, log
 
 from eval import evaluate
 from main import init, run
+from utils import load_image
 
 
 def load_images_from_dir(input_folder):
@@ -31,7 +32,7 @@ def load_images_from_dir(input_folder):
 
     images = []
     for path in paths:
-        image = zea.io_lib.load_image(path)
+        image = load_image(path)
         images.append(image)
 
     if len(images) == 0:

utils.py

@@ -1,5 +1,8 @@
+from pathlib import Path
+
 import numpy as np
 from keras import ops
+from PIL import Image
 from skimage import filters, morphology
 from zea.utils import translate
 
@@ -131,3 +134,30 @@ def extract_skeleton(images, input_range, sigma_pre=4, sigma_post=4, threshold=0
     skeleton_masks = (skeleton_masks - min_val) / (max_val - min_val + 1e-8)
 
     return ops.convert_to_tensor(skeleton_masks, dtype=images.dtype)
+
+
+def load_image(filename, grayscale=True):
+    """Load an image file and return a numpy array using PIL.
+
+    Args:
+        filename (str): The path to the image file.
+        grayscale (bool, optional): Whether to convert the image to grayscale. Defaults to True.
+
+    Returns:
+        numpy.ndarray: A numpy array of the image.
+
+    Raises:
+        FileNotFoundError: If the file does not exist.
+    """
+    filename = Path(filename)
+    if not filename.exists():
+        raise FileNotFoundError(f"File {filename} does not exist")
+
+    img = Image.open(filename)
+    if grayscale:
+        img = img.convert("L")
+    else:
+        img = img.convert("RGB")
+
+    arr = np.array(img)
+    return arr