flow to rgb fix

2025-05-01 16:43:28 -04:00
parent f591d43526
commit deceebb212
2 changed files with 3 additions and 3 deletions
@@ -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