mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-25 18:02:37 +08:00
Properly load higher bit depth videos. (#13542)
This commit is contained in:
parent
7636599389
commit
4304c15e9b
@ -248,8 +248,8 @@ class VideoFromFile(VideoInput):
|
||||
continue
|
||||
if self.__duration and frame.pts >= end_pts:
|
||||
break
|
||||
img = frame.to_ndarray(format='rgb24') # shape: (H, W, 3)
|
||||
img = torch.from_numpy(img) / 255.0 # shape: (H, W, 3)
|
||||
img = frame.to_ndarray(format='gbrpf32le') # shape: (H, W, 3)
|
||||
img = torch.from_numpy(img)
|
||||
frames.append(img)
|
||||
|
||||
images = torch.stack(frames) if len(frames) > 0 else torch.zeros(0, 3, 0, 0)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user