Como enviar imagens de vídeo OpenCV sobre soquetes ZeroMQ?
Eu tenho uma webcam simples que li usando o OpenCV e agora estou tentando enviar esse vídeo para um programa (Python) diferente usandoZeroMQ. Então, eu tenho o seguinte script simples para ler a webcam e enviá-la usando um soquete ZeroMQ:
import cv2
import os
import zmq
import base64
context = zmq.Context()
footage_socket = context.socket(zmq.PUB)
footage_socket.connect('tcp://localhost:5555')
# init the camera
camera = cv2.VideoCapture(0)
while True:
try:
(grabbed, frame) = camera.read() # grab the current frame
frame = cv2.resize(frame, (640, 480)) # resize the frame
footage_socket.send_string(base64.b64encode(frame))
# Show the video in a window
cv2.imshow("Frame", frame) # show the frame to our screen
cv2.waitKey(1) # Display it at least one ms
# # before going to the next frame
except KeyboardInterrupt:
camera.release()
cv2.destroyAllWindows()
print "\n\nBye bye\n"
break
Isso funciona bem, pois mostra o vídeo e não gera erros.
Comentei as duas linhas que mostram a imagem (cv2.imshow()
ecv2.waitKey(1)
) Comecei o script abaixo em paralelo. Este segundo script deve receber o vídeo e mostrá-lo.
import cv2
import zmq
import base64
import numpy as np
context = zmq.Context()
footage_socket = context.socket(zmq.SUB)
footage_socket.bind('tcp://*:5555')
footage_socket.setsockopt_string(zmq.SUBSCRIBE, unicode(''))
# camera = cv2.VideoCapture("output.avi")
while True:
try:
frame = footage_socket.recv_string()
frame = np.fromstring(base64.b64decode(frame), dtype=np.uint8)
cv2.imshow("Frame", frame) # show the frame to our screen
cv2.waitKey(1) # Display it at least one ms
# # before going to the next frame
except KeyboardInterrupt:
cv2.destroyAllWindows()
break
print "\n\nBye bye\n"
Infelizmente, isso congela emcv2.waitKey(1)
.
Alguém sabe o que estou fazendo de errado aqui? Preciso decodificar a filmagem de maneira diferente? Todas as dicas são bem-vindas!