Function _count_video_frames

tools/preprocess/ltx2.3_preprocess_data.py:276–284

Count the actual decodable frame count of an mp4 (PyAV demux).

(path: str)

def _count_video_frames(path: str) -> int:
    """Count the actual decodable frame count of an mp4 (PyAV demux)."""
    c = av.open(path)
    try:
        s = next(x for x in c.streams if x.type == "video")
        n = sum(1 for _ in c.decode(s))
    finally:
        c.close()
    return n


def _save_video(frames_rgb: List[np.ndarray], out_path: str, fps: float):