Files
project_infer/lane_server/test_camera_recv.py
hkslover 5840e5883f 5.24
2024-05-24 13:00:10 +08:00

20 lines
433 B
Python

import zmq
import numpy as np
import cv2
context = zmq.Context()
socket = context.socket(zmq.SUB)
socket.connect("tcp://localhost:5555")
socket.setsockopt_string(zmq.SUBSCRIBE, '')
while True:
message = socket.recv()
np_array = np.frombuffer(message, dtype=np.uint8)
frame = cv2.imdecode(np_array, cv2.IMREAD_COLOR)
cv2.imshow("Received", frame)
if cv2.waitKey(1) == 27:
break
cv2.destroyAllWindows()