Поменял, метод формирования видео

main.py

@@ -98,9 +98,8 @@ class InterpolationPipeline:
 
     def run(self, video_path: Path, output_video: str):
         prev_frames = tuple()
-        interpolated_frames = []
+        interpolated_frames: list["np.ndarray"] = []
         part = 0
-        source_frame_length = 0
         chunk_seconds = 10
         length = self.video_maker.get_video_duration(video_path)
         last_part_seconds = 1 if length % chunk_seconds else 0
@@ -108,6 +107,7 @@ class InterpolationPipeline:
         fps = self.video_maker.get_fps(video_path)
         logging.info(f"Video FPS: {fps}")
         fps *= 2  # Doubling FPS
+        width, height = self.video_maker.get_size(video_path)
         for frames in self.video_maker.video_to_frames_generator(
             video_path, self.fs.frames_path, chunk_seconds
         ):
@@ -117,13 +117,10 @@ class InterpolationPipeline:
                 img2 = frames[0]
                 img1_2 = self.interpolator.interpolate(img1, img2)
                 interpolated_frames.append(img1_2)
-                self.fs.clear_directory(self.fs.moved_path)
-                self._save_images(prev_frames, interpolated_frames)
-
-                self._merge_frames_to_video(
-                    self.fs.video_part_path / f"video_{part:08d}.mp4",
-                    fps,
-                    source_frame_length,
+                generator = self._frame_generator(prev_frames, interpolated_frames)
+                part_path = self.fs.video_part_path / f"video_{part:08d}.mp4"
+                self.video_maker.images_to_video_pipeline(
+                    generator, part_path, width, height, fps
                 )
                 interpolated_frames = []
                 logging.info(f"Finished processing part {part:08d}")
@@ -136,17 +133,13 @@ class InterpolationPipeline:
                 img2 = frames[i + 1]
                 img1_2 = self.interpolator.interpolate(img1, img2)
                 interpolated_frames.append(img1_2)
-            source_frame_length = len(frames)
             prev_frames = frames
 
-        self.fs.clear_directory(self.fs.moved_path)
-        self._save_images(prev_frames, interpolated_frames)
-        self._merge_frames_to_video(
-            self.fs.video_part_path / f"video_{part:08d}.mp4",
-            fps,
-            source_frame_length,
+        generator = self._frame_generator(prev_frames, interpolated_frames)
+        part_path = self.fs.video_part_path / f"video_{part:08d}.mp4"
+        self.video_maker.images_to_video_pipeline(
+            generator, part_path, width, height, fps
         )
-        self.fs.clear_directory(self.fs.moved_path)
         logging.info(f"Finished processing part {part:08d}")
         self._merge_video_parts(self.fs.output_path / output_video)
         logging.info(
@@ -159,6 +152,7 @@ class InterpolationPipeline:
         interpolated: list["np.ndarray"],
     ):
         logging.info("Saving images...")
+        self.fs.clear_directory(self.fs.moved_path)
         index = 0
         for i, frame in enumerate(source):
             name = self.fs.moved_path / f"img_{index:08d}.png"
@@ -170,15 +164,23 @@ class InterpolationPipeline:
                 imwrite(name, interpolated[i])
         logging.info("Success...")
 
-    def _merge_frames_to_video(
-        self, output_video: Path, fps: float, source_frame_length: int = 0
-    ):
+    def _merge_frames_to_video(self, output_video: Path, fps: float):
         self.video_maker.images_to_video(self.fs.moved_path, output_video, fps)
 
     def _merge_video_parts(self, output_video: Path):
         self.video_maker.concatenate_videos(self.fs.video_part_path, output_video)
         self.fs.clear_directory(self.fs.video_part_path)
 
+    def _frame_generator(
+        self,
+        source: tuple["np.ndarray", ...],
+        interpolated: list["np.ndarray"],
+    ):
+        for i, frame in enumerate(source):
+            yield frame
+            if i < len(interpolated):
+                yield interpolated[i]
+
 
 def runner(
     base_path: Path,