Перенес networks внутрь src

main.py

@@ -1,11 +1,11 @@
 import logging
 import subprocess
 from pathlib import Path
+from typing import Generator
 
 import cv2
 from tqdm import tqdm
 from time import perf_counter
-from decimal import Decimal
 
 from interpolator import get_device
 from interpolator import ImageInterpolator
@@ -39,68 +39,11 @@ def move_images(src_dir: str, interpolated_dir: str, output_dir: str):
             index += 1
 
 
-def build_file_list(moved_dir: str, list_path: str):
-    import os
-    moved_dir = Path(moved_dir)
-    frames = sorted(moved_dir.glob("img_*.png"))
-    print(frames[0])
-
-    with open(list_path, "w") as f:
-        for frame in frames:
-            f.write(f"file '{os.path.abspath(frame)}'\n")
-
-
-def build_ffmpeg_file_list(frames_dir: str, interpolated_dir: str, list_path: str):
-    frames = sorted(Path(frames_dir).glob("img_*.png"))
-    interps = sorted(Path(interpolated_dir).glob("img_*.png"))
-
-    if len(interps) != len(frames) - 1:
-        raise ValueError("Interpolated frames must be N-1")
-
-    with open(list_path, "w") as f:
-        for i in range(len(frames)):
-            f.write(f"file '{frames[i].resolve().as_posix()}'\n")
-
-            if i < len(interps):
-                f.write(f"file '{interps[i].resolve().as_posix()}'\n")
-
-
-def merge_with_ffmpeg(
-    original_video: str,
-    file_list: str,
-    output_video: str,
-):
-    cap = cv2.VideoCapture(original_video)
-
-    if not cap.isOpened():
-        raise ValueError("Cannot open original video")
-
-    fps = cap.get(cv2.CAP_PROP_FPS)
-    cap.release()
-
-    new_fps = Decimal(fps * 2)
-
-    cmd = [
-        "ffmpeg",
-        "-y",
-        "-r", str(new_fps.quantize(Decimal("1.0000000000"))),
-        "-f", "concat",
-        "-safe", "0",
-        "-i", file_list,
-        "-c:v", "libx264rgb",
-        output_video,
-    ]
-    print("Running ffmpeg command:", " ".join(cmd))
-
-    subprocess.run(cmd, check=True)
-
-
-
 def video_frames_to_disk_generator(
     video_path: str | Path,
     output_dir: str | Path,
     chunk_seconds: int = 10
-):
+) -> Generator[tuple[Path, ...], None, None]:
     output_dir = Path(output_dir)
     output_dir.mkdir(parents=True, exist_ok=True)