Initial Commit

2025-04-09 19:44:42 +01:00
parent b6ed1ef3c8
commit 13487afa05
3 changed files with 44 additions and 0 deletions
@@ -0,0 +1,34 @@
+import requests
+from PIL import Image
+import io
+import numpy as np
+import torch
+
+class LoadImageFromHttpURL:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "image_url": ("STRING", {
+                    "default": "https://placehold.co/600x400/EEE/31343C"
+                }),
+            }
+        }
+
+    RETURN_TYPES = ("IMAGE",)
+    FUNCTION = "load_image"
+    CATEGORY = "jerrywap/loaders"
+
+    def load_image(self, image_url: str):
+        try:
+            response = requests.get(image_url)
+            response.raise_for_status()
+            image = Image.open(io.BytesIO(response.content)).convert("RGB")
+
+            image_np = np.array(image).astype(np.float32) / 255.0
+            image_tensor = torch.from_numpy(image_np).unsqueeze(0)
+
+            return (image_tensor,)
+        except Exception as e:
+            print(f"Failed to load image from URL: {e}")
+            return (torch.zeros(1, 512, 512, 3),)  # fallback black image