/
g-konst
/
zebra_test
Обзор
Документация
Войти
/
g-konst
/
zebra_test
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
scripts/extract.py
35 строк
911 B
Konstantin Grudnitskiy
refactored
25 июн 2025, 13:36
Не верифицирован
25 июн 2025, 13:36
487a489
Код
Авторство
О чём код?
from pathlib import Path import cv2 INPUT_DIR = Path("inputs") OUTPUT_DIR = INPUT_DIR / "images" def extract_frames(video, fps: int = 5): count, saved = 0, 0 cap = cv2.VideoCapture(video) framerate = int(cap.get(cv2.CAP_PROP_FPS)) print("Video framerate:", framerate) while cap.isOpened(): success, image = cap.read() if not success: break if count % (framerate / fps) == 0: filename = OUTPUT_DIR / f"frame_{saved:05d}.jpg" cv2.imwrite(filename, image) saved += 1 count += 1 cap.release() print(f"Saved {saved!r} of {count!r} frames.") if __name__ == "__main__": OUTPUT_DIR.mkdir(parents="True", exist_ok=True) for video in (INPUT_DIR / "videos").glob("*.mov", case_sensitive=False): print("Extracting frames from video:", video) extract_frames(video) print("done")