Files
project_infer/lane_server/test_camera_recv.py

20 lines
433 B
Python
Raw Normal View History

2024-05-24 13:00:10 +08:00
import zmq
import numpy as np
import cv2
context = zmq.Context()
socket = context.socket(zmq.SUB)
2024-06-07 20:19:04 +08:00
socket.connect("tcp://localhost:5556")
2024-05-24 13:00:10 +08:00
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()