quochuy commited on
Commit
b01f428
·
1 Parent(s): 54ea07d

update tcp

Browse files
Files changed (1) hide show
  1. inference.py +3 -1
inference.py CHANGED
@@ -20,17 +20,19 @@ def get_init_states():
20
  dummy = tf.zeros([1, 1, RESOLUTION, RESOLUTION, 3], dtype=tf.float32)
21
  return model.init_states(tf.shape(dummy))
22
  os.environ["OPENCV_FFMPEG_INTERRUPT_TIMEOUT"] = "60000"
 
23
  class VideoProcessor:
24
  def __init__(self):
25
  self.running = False
26
 
27
  def start_processing(self, rtsp_url, result_queue):
28
  self.running = True
 
29
  cap = cv2.VideoCapture(rtsp_url, cv2.CAP_FFMPEG)
30
  if not cap.isOpened():
31
  result_queue.put({"error": "Cannot open RTSP URL"})
32
  return
33
-
34
  states = get_init_states()
35
 
36
  # Logic quản lý sự kiện (Cooldown)
 
20
  dummy = tf.zeros([1, 1, RESOLUTION, RESOLUTION, 3], dtype=tf.float32)
21
  return model.init_states(tf.shape(dummy))
22
  os.environ["OPENCV_FFMPEG_INTERRUPT_TIMEOUT"] = "60000"
23
+ os.environ["OPENCV_FFMPEG_CAPTURE_OPTIONS"] = "rtsp_transport;tcp"
24
  class VideoProcessor:
25
  def __init__(self):
26
  self.running = False
27
 
28
  def start_processing(self, rtsp_url, result_queue):
29
  self.running = True
30
+ print(f"Trying to open: {rtsp_url}")
31
  cap = cv2.VideoCapture(rtsp_url, cv2.CAP_FFMPEG)
32
  if not cap.isOpened():
33
  result_queue.put({"error": "Cannot open RTSP URL"})
34
  return
35
+ print("RTSP Stream Opened successfully with TCP!")
36
  states = get_init_states()
37
 
38
  # Logic quản lý sự kiện (Cooldown)