From deceebb2128a42e3b6c7dc15efe3340d9503f8a8 Mon Sep 17 00:00:00 2001
From: RyanOnTheInside <7623207+ryanontheinside@users.noreply.github.com>
Date: Thu, 1 May 2025 16:43:28 -0400
Subject: [PATCH] flow to rgb fix

---
 node_wrappers/realtimenodes/utils/temporal_processing.py | 4 ++--
 src/utils/image.py                                       | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/node_wrappers/realtimenodes/utils/temporal_processing.py b/node_wrappers/realtimenodes/utils/temporal_processing.py
index 39b3c0a..2b3a77c 100644
--- a/node_wrappers/realtimenodes/utils/temporal_processing.py
+++ b/node_wrappers/realtimenodes/utils/temporal_processing.py
@@ -2,7 +2,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import math
-from ....src.utils.image import gaussian_blur_2d
+from ....src.utils.image import gaussian_blur_2d, flow_to_rgb
 from ....src.utils.realtime_flownets import RealTimeFlowNet
 #NOTE: this is totally experimental and grounded in very little research.
 
@@ -52,7 +52,7 @@ class TemporalNetV2Preprocessor:
         
         # Visualize flow as RGB (similar to flow_to_image in torchvision)
         # This converts 2-channel flow to 3-channel RGB visualization
-        flow_rgb = self.flow_to_rgb(flow)
+        flow_rgb = flow_to_rgb(flow)
         
         # Convert flow back to BHWC format
         flow_rgb = flow_rgb.movedim(1, -1)
diff --git a/src/utils/image.py b/src/utils/image.py
index f223f2d..b412ab5 100644
--- a/src/utils/image.py
+++ b/src/utils/image.py
@@ -14,7 +14,7 @@ import math
 import torch
 import torch.nn.functional as F
 
-def flow_to_rgb(self, flow):
+def flow_to_rgb(flow):
     """
     Convert optical flow to RGB visualization similar to torchvision's flow_to_image