zmq client_test.py

import cv2
import zmq

# from djitellopy import Tello
# tello gogo
import time


SERVER_IP = "cam.uhmcv.kro.kr"
PORT = 5000

print("create ZMQ")
context = zmq.Context()
socket = context.socket(zmq.PUSH)
socket.connect(f"tcp://{SERVER_IP}:{PORT}")

print(f"try to connect {SERVER_IP}:{PORT}")

cap = cv2.VideoCapture(0)

if not cap.isOpened():
    print("error: cannot open webcam")
    exit()


# cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640)
# cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)

try:
    while True:
        ret, frame = cap.read()
        if not ret or frame is None:
            print("error reading frame from webcam")
            continue

        ret_encode, buffer = cv2.imencode(".jpg", frame, [cv2.IMWRITE_JPEG_QUALITY, 80])
        if not ret_encode:
            print("encode error")
            continue

        socket.send(buffer)

        # cv2.imshow("Webcam Stream - Laptop", frame)
        if cv2.waitKey(1) & 0xFF == ord("q"):
            break

        time.sleep(0.01)

except KeyboardInterrupt:
    print("stop service")

finally:
    cap.release()
    socket.close()
    context.term()
    cv2.destroyAllWindows()