1
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00

Add get_duration method to Comfy VIDEO type (#8122)

* get duration from VIDEO type

* video get_duration unit test

* fix Windows unit test: can't delete opened temp file
This commit is contained in:
Christian Byrne
2025-05-14 21:11:41 -07:00
committed by GitHub
parent 08368f8e00
commit f1f9763b4c
3 changed files with 281 additions and 0 deletions

View File

@@ -43,3 +43,13 @@ class VideoInput(ABC):
components = self.get_components()
return components.images.shape[2], components.images.shape[1]
def get_duration(self) -> float:
"""
Returns the duration of the video in seconds.
Returns:
Duration in seconds
"""
components = self.get_components()
frame_count = components.images.shape[0]
return float(frame_count / components.frame_rate)