Improve duration estimate

NINAnor · Jul 30, 2024 · ae2c504 · ae2c504
1 parent 1a3e95c
commit ae2c504
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/app/data_manager/data_manager.py b/app/data_manager/data_manager.py
@@ -199,9 +199,8 @@ def get_video_duration(self, path: Path) -> str:
         """
         try:
             cap = cv2.VideoCapture(str(path))
-            fps = cap.get(cv2.CAP_PROP_FPS)
-            frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-            duration = frame_count / fps
+            cap.set(cv2.CAP_PROP_POS_AVI_RATIO, 1)
+            duration = cap.get(cv2.CAP_PROP_POS_MSEC)
             return str(datetime.timedelta(seconds=duration))
         except cv2.error as error:
             logger.error("Error occurred: %s", error)