bgr to rgb format when using cv2

lucidrains · Nov 27, 2023 · 286d4de · 286d4de
1 parent dfd79cf
commit 286d4de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/magvit2_pytorch/data.py b/magvit2_pytorch/data.py
@@ -184,6 +184,7 @@ def video_to_tensor(
     frames_torch = torch.tensor(frames).float()
 
     frames_torch /= 255.
+    frames_torch = frames_torch.flip(dims = (0,)) # BGR -> RGB format
 
     return frames_torch[:, :num_frames, :, :]
 

diff --git a/magvit2_pytorch/version.py b/magvit2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.1.41'
+__version__ = '0.1.42'