From ca1333bafb3b25d06fcbb097087c987fa60f038b Mon Sep 17 00:00:00 2001
From: luanhailiang <hi@luanhailiang.cn>
Date: Mon, 25 Sep 2023 14:51:14 +0800
Subject: [PATCH] sleep for mp4

---
 app.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/app.py b/app.py
index a7bba1c..c706040 100644
--- a/app.py
+++ b/app.py
@@ -67,12 +67,21 @@ def worker(id,source,region=None,stream=False):
     print("start loaded  model!!!")
     balls = {}
     count = 0
+    delay = 0
+    bgn = 0
     try:
         cap = cv2.VideoCapture(source)
         if not cap.isOpened():
             print("Error opening video stream.")
+        if source.endswith(".mp4"):
+            fps = cap.get(cv2.CAP_PROP_FPS)
+            delay = 1000.0 / fps
+            print("start video stream... ", fps, delay)
         while True:
             use = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+            if bgn != 0 and delay != 0 and time.time()*1000 - bgn < delay:
+                print("sleep",delay - (time.time()*1000 - bgn))
+                time.sleep((delay - (time.time()*1000 - bgn))/1000)
             bgn = (time.time()*1000)
             ret, frame = cap.read()
             del(ret)