From bb5092f7008cf74d8a9d2b465f22381fe41f65f7 Mon Sep 17 00:00:00 2001
From: Max <m.giller.dev@gmail.com>
Date: Sat, 6 Sep 2025 19:15:43 +0200
Subject: [PATCH] Added missing reference normalization

---
 src/label_episodes.py | 56 +++++++++++++++++++++----------------------
 1 file changed, 28 insertions(+), 28 deletions(-)

diff --git a/src/label_episodes.py b/src/label_episodes.py
index 3c085ee..5016e49 100644
--- a/src/label_episodes.py
+++ b/src/label_episodes.py
@@ -230,53 +230,53 @@ if __name__ == "__main__":
         # "./episodes/B1_t09.mkv",
         # "./episodes/B1_t10.mkv",
     ]
-    
+
     references = {
-        "1" : [
-            cv2.imread("./1/9hYDr40VIWbAHGjmlpSkyqlDaVQ.webp"),
-            cv2.imread("./1/cmYUHroQ9zRhOYiTf0ozMWe1Bzs.webp"),
-            cv2.imread("./1/yLwaP7Y7O3Wyrht9lPmTDk7LSuG.webp"),
+        "1": [
+            normalize_image(cv2.imread("./1/9hYDr40VIWbAHGjmlpSkyqlDaVQ.webp")),
+            normalize_image(cv2.imread("./1/cmYUHroQ9zRhOYiTf0ozMWe1Bzs.webp")),
+            normalize_image(cv2.imread("./1/yLwaP7Y7O3Wyrht9lPmTDk7LSuG.webp")),
         ],
-        "2" : [
-            cv2.imread("./2/g2pyyFRFBS18jYByRdiunWr7nap.webp"),
-            cv2.imread("./2/wKCm3I7efXfsWGNrG28d3cIsL9B.webp"),
-            cv2.imread("./2/yhL1JBnTDK30BtPbr8gZ5ZXCASL.webp"),
+        "2": [
+            normalize_image(cv2.imread("./2/g2pyyFRFBS18jYByRdiunWr7nap.webp")),
+            normalize_image(cv2.imread("./2/wKCm3I7efXfsWGNrG28d3cIsL9B.webp")),
+            normalize_image(cv2.imread("./2/yhL1JBnTDK30BtPbr8gZ5ZXCASL.webp")),
         ],
-        "3" : [
-            cv2.imread("./3/d9F2x01XNhi65GTXwQjrdDJKgqm.webp"),
-            cv2.imread("./3/kV7VAvLgjjhPTI56OnNvcNaHjDM.webp"),
-            cv2.imread("./3/mLxkumagHSYg4KNLcEIgHpiJrx.webp"),
+        "3": [
+            normalize_image(cv2.imread("./3/d9F2x01XNhi65GTXwQjrdDJKgqm.webp")),
+            normalize_image(cv2.imread("./3/kV7VAvLgjjhPTI56OnNvcNaHjDM.webp")),
+            normalize_image(cv2.imread("./3/mLxkumagHSYg4KNLcEIgHpiJrx.webp")),
         ],
-        "4" : [
-            cv2.imread("./4/9q62zw89SJiRD3kjNuhUIswWHTa.webp"),
-            cv2.imread("./4/lyi4XUUssCacaEQs2FCGSDCKDUK.webp"),
+        "4": [
+            normalize_image(cv2.imread("./4/9q62zw89SJiRD3kjNuhUIswWHTa.webp")),
+            normalize_image(cv2.imread("./4/lyi4XUUssCacaEQs2FCGSDCKDUK.webp")),
         ],
-        "5" : [
-            cv2.imread("./5/uIztge10KNlRU4KUFNXWPGCiG8e.webp"),
-            cv2.imread("./5/vuhdflodFfPkvekVWbS4JZMYHa7.webp"),
-            cv2.imread("./5/xmvi4bBk5TQyuL6tpiYvNoIPft.webp"),
+        "5": [
+            normalize_image(cv2.imread("./5/uIztge10KNlRU4KUFNXWPGCiG8e.webp")),
+            normalize_image(cv2.imread("./5/vuhdflodFfPkvekVWbS4JZMYHa7.webp")),
+            normalize_image(cv2.imread("./5/xmvi4bBk5TQyuL6tpiYvNoIPft.webp")),
         ],
-        "6" : [
-            cv2.imread("./6/2I7WgvRwnYDNjbiztvbOAldYGde.webp"),
-            cv2.imread("./6/dIaBagmUbP1AO6xFQzfQud5lk4I.webp"),
-            cv2.imread("./6/ekDZrFRDJDyx78uttMyVT9mPhjL.webp"),
+        "6": [
+            normalize_image(cv2.imread("./6/2I7WgvRwnYDNjbiztvbOAldYGde.webp")),
+            normalize_image(cv2.imread("./6/dIaBagmUbP1AO6xFQzfQud5lk4I.webp")),
+            normalize_image(cv2.imread("./6/ekDZrFRDJDyx78uttMyVT9mPhjL.webp")),
         ],
     }
 
     results = match_episodes_to_references(episodes, references)
-    
+
     print(results)
-    
+
     print("\nBy Episode:")
     for ep, scores in results.items():
         print(f"{ep}")
         for ref, value in scores.items():
             print(f"\t{ref}\t{value}")
-            
+
     print("\nBy Reference:")
     for ref in list(results.values())[0].keys():
         print(f"{ref}")
         for episode, scores in results.items():
             print(f"\t{episode}\t{scores[ref]}")
-    
+
     print("Done.")