changed algorithm for difference generation to a much faster one

Author: TheElevatedOne

.gitignore

@@ -1,3 +1,6 @@
 venv
 main.build
 dist
+test
+1x-WB-Denoise.pth
+test.py

differ/__pycache__/differ_gen.cpython-310.pyc

@@ -1,7 +1,7 @@
 import argparse
 import multiprocessing
 import sys
-from differ.differ import Difference
+from src.differ import Difference
 
 
 def parse() -> argparse.Namespace:

differ/differ_gen.py

@@ -1,7 +1,8 @@
-blend_modes==2.2.0
-numpy==2.2.0
-opencv-python==4.10.0.84
-pillow==11.0.0
-pycuda==2024.1.2
-torch==2.5.1
-tqdm==4.67.1
+numpy>=2.2.0
+opencv-python>=4.10.0.84
+pillow>=11.0.0
+pycuda>=2024.1.2
+torch>=2.5.1
+tqdm>=4.67.1
+imageio>=2.36.1
+nuitka>=2.5.8

main.py

@@ -0,0 +1,35 @@
+import os
+import numpy as np
+from queue import Queue
+import imageio.v3 as imio3
+
+class DifferGen:
+    def __init__(self, temp_dir: str) -> None:
+        """
+        class DifferGen requires Temporary Directory path as an Initialization Variable
+        """
+        self.tdir = temp_dir
+
+    def run(self, imgs: list, queue: Queue) -> None:
+        for img in imgs:
+            bt, tp = os.path.join(self.tdir, "bottom", img), os.path.join(self.tdir, "top", img)
+            # bottom_raw = Image.open(bt).convert("RGBA")
+            # top_raw = Image.open(tp).convert("RGBA")
+            #
+            # bottom_npy = numpy.array(bottom_raw)
+            # top_npy = numpy.array(top_raw)
+            #
+            # bottom_fl = bottom_npy.astype(float)
+            # top_fl = top_npy.astype(float)
+            #
+            # diff_fl = difference(bottom_fl, top_fl, 1)
+            # diff_npy = numpy.uint8(diff_fl)
+            # diff_raw = Image.fromarray(diff_npy)
+            #
+            # diff_raw.save(os.path.join(self.tdir, "diff", img))
+
+            bottom, top = np.int16(imio3.imread(bt)), np.int16(imio3.imread(tp))
+            diff = np.uint8(np.abs(bottom - top))
+            imio3.imwrite(os.path.join(self.tdir, "diff", img), diff)
+
+            queue.put(0)