quochuy
commited on
Commit
·
b01f428
1
Parent(s):
54ea07d
update tcp
Browse files- 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)
|