blakeblackshear пре 6 година
родитељ
комит
aeece7b98c
1 измењених фајлова са 8 додато и 1 уклоњено
  1. 8 1
      detect_objects.py

+ 8 - 1
detect_objects.py

@@ -360,11 +360,18 @@ def fetch_frames(shared_arr, shared_frame_time, frame_lock, frame_ready, frame_s
     arr = tonumpyarray(shared_arr).reshape(frame_shape)
 
     # start the video capture
-    video = cv2.VideoCapture(RTSP_URL)
+    video = cv2.VideoCapture()
+    video.open(RTSP_URL)
     # keep the buffer small so we minimize old data
     video.set(cv2.CAP_PROP_BUFFERSIZE,1)
 
     while True:
+        # check if the video stream is still open, and reopen if needed
+        if not video.isOpened():
+            success = video.open(RTSP_URL)
+            if not success:
+                time.sleep(1)
+                continue
         # grab the frame, but dont decode it yet
         ret = video.grab()
         # snapshot the time the frame was grabbed