From 41990b0228c1ba158c396a3b65dafaa6aa24983f Mon Sep 17 00:00:00 2001 From: u Date: Mon, 9 Sep 2024 01:10:00 +0800 Subject: [PATCH] # BGR to RGB --- inference/gradio_composite_demo/rife_model.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/inference/gradio_composite_demo/rife_model.py b/inference/gradio_composite_demo/rife_model.py index b7eac2b..f66d5e7 100644 --- a/inference/gradio_composite_demo/rife_model.py +++ b/inference/gradio_composite_demo/rife_model.py @@ -103,9 +103,13 @@ def rife_inference_with_path(model, video_path): pt_frame_data = [] pt_frame = skvideo.io.vreader(video_path) for frame in pt_frame: + # BGR to RGB + frame_rgb = frame[..., ::-1] + frame_rgb = frame_rgb.copy() + tensor = torch.from_numpy(frame_rgb).float().to("cpu", non_blocking=True).float() / 255.0 pt_frame_data.append( - torch.from_numpy(np.transpose(frame, (2, 0, 1))).to("cpu", non_blocking=True).float() / 255.0 - ) + tensor.permute(2, 0, 1) + ) # to [c, h, w,] pt_frame = torch.from_numpy(np.stack(pt_frame_data)) pt_frame = pt_frame.to(device)