Update utils.py

utils.py

@@ -1,5 +1,6 @@
 import numpy as np
 import cv2
+from PIL import Image
 
 def write_video(file_path, frames, fps):
     """
@@ -23,3 +24,22 @@ def write_video(file_path, frames, fps):
 
 def dummy(images, **kwargs):
     return images, False
+
+def preprocess_image(current_image, steps, image_size):
+    next_image = np.array(current_image.convert("RGBA"))*0
+    prev_image = current_image.resize((image_size-2*steps,image_size-2*steps))
+    prev_image = prev_image.convert("RGBA")
+    prev_image = np.array(prev_image)
+    next_image[:, :, 3] = 1
+    next_image[steps:image_size-steps,steps:image_size-steps,:] = prev_image
+    prev_image = Image.fromarray(next_image)
+    
+    return prev_image
+
+
+def preprocess_mask_image(current_image):
+    mask_image = np.array(current_image)[:,:,3] # assume image has alpha mask (use .mode to check for "RGBA")
+    mask_image = Image.fromarray(255-mask_image).convert("RGB")
+    current_image = current_image.convert("RGB")
+    
+    return current_image, mask_image