update: Video2ImageProcessor의 call을 동기식으로 변경

404Vector · Nov 28, 2023 · fc1b926 · fc1b926
1 parent 118461e
commit fc1b926
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/v2v/_video2image_processor_.py b/v2v/_video2image_processor_.py
@@ -24,23 +24,20 @@ def __init__(self, video_path: str, ffmpeg_options_output: Optional[dict] = None
         self._progress = 0
         self._is_done = False
 
-    async def __call__(self) -> FrameData:
+    def __call__(self) -> FrameData:
         assert self._is_done is False
-        frame = await asyncio.to_thread(
-            functools.partial(
-                read_frame_from_process,
-                process=self._sub_processor,
-                width=self._video_info.frame_width,
-                height=self._video_info.frame_height,
-            )
+        frame = read_frame_from_process(
+            process=self._sub_processor,
+            width=self._video_info.frame_width,
+            height=self._video_info.frame_height,
         )
 
         if frame is not None:
             frame_data = FrameData(frame_id=self._progress, frame=frame)
             self._progress += 1
         else:
             self._is_done = True
-            frame_data = FrameData(frame_id=-1, frame=None)
+            frame_data = None
             self._sub_processor.stdout.close()
             self._sub_processor.wait()
         return frame_data