Add helper function

modules/live_portrait/live_portrait_inferencer.py

@@ -550,6 +550,9 @@ class LivePortraitInferencer:
         # source_image_np = (source_image * 255).byte().numpy()
         # img_rgb = source_image_np[0]
         # print("Prepare source...")
+        if isinstance(source_image, str):
+            source_image = image_path_to_array(source_image)
+
         if len(source_image.shape) <= 3:
             source_image = source_image[np.newaxis, ...]
 

modules/utils/image_helper.py

@@ -56,3 +56,12 @@ def calc_crop_limit(center, img_size, crop_size):
 def save_image(numpy_array: np.ndarray, output_path: str):
     out = Image.fromarray(numpy_array)
     out.save(output_path, compress_level=1, format="png")
+
+
+def image_path_to_array(image_path: str) -> np.ndarray:
+    image = Image.open(image_path)
+    image_array = np.array(image)
+    if len(image_array.shape) <= 3:
+        image_array = image_array[np.newaxis, ...]
+
+    return image_array