wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-01 01:00:20 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Count packets unless codec has subframes

Browse Source
This commit is contained in:
Austin Mroz 2026-01-28 23:15:06 -08:00
parent 7d2c369f45
commit ae7bf48331
No known key found for this signature in database
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
comfy_api/latest/_input_impl/video_types.py
View File

@ -120,7 +120,12 @@ class VideoFromFile(VideoInput):
if video_stream and video_stream.average_rate: if video_stream and video_stream.average_rate:
frame_count = 0 frame_count = 0
container.seek(0) container.seek(0)
for packet in container.demux(video_stream): frame_iterator = (
container.decode(video_stream)
if video_stream.codec.capabilities & 0x100
else container.demux(video_stream)
)
for packet in frame_iterator:
for _ in packet.decode(): for _ in packet.decode():
frame_count += 1 frame_count += 1
if frame_count > 0: if frame_count > 0:
@ -165,7 +170,11 @@ class VideoFromFile(VideoInput):
frame_count = 1 frame_count = 1
start_pts = int(self.__start_time / video_stream.time_base) start_pts = int(self.__start_time / video_stream.time_base)
container.seek(start_pts, stream=video_stream) container.seek(start_pts, stream=video_stream)
frame_iterator = container.decode(video_stream) frame_iterator = (
container.decode(video_stream)
if video_stream.codec.capabilities & 0x100
else container.demux(video_stream)
)
for frame in frame_iterator: for frame in frame_iterator:
if frame.pts >= start_pts: if frame.pts >= start_pts:
break break