Update forensics/ela_hybrid.py

forensics/ela_hybrid.py

@@ -50,12 +50,14 @@ def save_hybrid_image(hybrid_array, save_path: str):
     np.save(save_path, hybrid_array)
     print(f"Saved hybrid ELA image to {save_path}")
 
+def visualize_hybrid(hybrid_array: np.ndarray):
+    """Return a list of 2 images: [RGB Image, ELA Map (grayscale)], as PIL Images."""
+    from PIL import Image
 
-def visualize_hybrid(hybrid_array: np.ndarray, split_visualize: bool = True):
-    """Split the 6 channels into RGB and ELA for visualization."""
-    rgb_image = (hybrid_array[:, :, :3] * 255).astype(np.uint8)
-    ela_image = (hybrid_array[:, :, 3:] * 255).astype(np.uint8)
-    return {
-        "rgb": Image.fromarray(rgb_image),
-        "ela": Image.fromarray(ela_image),
-    }
+    # Extract RGB (3 channels at front)
+    rgb_image = Image.fromarray((hybrid_array[:, :, :3] * 255).astype(np.uint8))
+    
+    # Extract ELA channels (last 3)
+    ela_image = Image.fromarray((hybrid_array[:, :, 3:] * 255).astype(np.uint8))
+
+    return [rgb_image, ela_image]