Update mineral.py

Codebugged-Research · Dec 7, 2023 · daf25b6 · daf25b6
1 parent efe4e8c
commit daf25b6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/margsoft/mineral.py b/src/margsoft/mineral.py
@@ -18,8 +18,9 @@ def get_video_capture(rtsp_url):
             print("Failed to open the stream.")
             return None
         return cap
-
-    cap = get_video_capture(rtsp)
+
+    cap = cv2.VideoCapture(rtsp, cv2.CAP_FFMPEG)
+    # cap = get_video_capture(rtsp)
     class_names = []
     with open(names_file, "r") as f:
         class_names = [cname.strip() for cname in f.readlines()]
@@ -65,6 +66,7 @@ def get_video_capture(rtsp_url):
                     cap = get_video_capture(rtsp)
             else:
                 # Attempt to reconnect if the capture is None
+                print("Error in frame")
                 time.sleep(3)
                 cap = get_video_capture(rtsp)