From 60ad56bb711dd32b88d8fc985d34a650cd209632 Mon Sep 17 00:00:00 2001
From: spacepxl <spacetug93@gmail.com>
Date: Sat, 3 Aug 2024 19:08:02 -0400
Subject: [PATCH] added MAE metric to DifferenceChecker

---
 nodes.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/nodes.py b/nodes.py
index 64a9e76..66bf25e 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1099,18 +1099,21 @@ class DifferenceChecker:
                 "images1": ("IMAGE", ),
                 "images2": ("IMAGE", ),
                 "multiplier": ("FLOAT", {"default": 1.0, "min": 0.01, "max": 1000.0, "step": 0.01,  "round": 0.01}),
+                "print_MAE": ("BOOLEAN", {"default": False}),
             },
         }
 
     RETURN_TYPES = ("IMAGE",)
     FUNCTION = "difference_checker"
-
+    OUTPUT_NODE = True
     CATEGORY = "image/filters"
 
-    def difference_checker(self, images1, images2, multiplier):
+    def difference_checker(self, images1, images2, multiplier, print_MAE):
         t = copy.deepcopy(images1)
-        t = torch.abs(images1 - images2) * multiplier
-        return (torch.clamp(t, min=0, max=1),)
+        t = torch.abs(images1 - images2)
+        if print_MAE:
+            print(f"MAE = {torch.mean(t)}")
+        return (torch.clamp(t * multiplier, min=0, max=1),)
 
 class ImageConstant:
     def __init__(self, device="cpu"):