图像通过TCP发送

时间:2013-12-28 23:45:25

标签: python opencv tcp raspberry-pi

我正在尝试使用python(raspberry PI)通过TCP Socket发送捕获图像的数据。 TCP服务器(QT)在其他计算机上运行并尝试读取图像并在QLabel中显示。

我从未在python中编程,完全不明白我做错了什么。我花了很多时间(天)来弥补正在进行的错误,最终达到了我可以实际运行它的程度。但我收到的数据就是垃圾。

以下程序在覆盆子PI的sep窗口中显示视频,并应通过套接字发送单个捕获的图像。

import cv2.cv as cv
import cv2
import time
from socket import socket
import sys
import numpy

cv.NamedWindow("camera",cv.CV_WINDOW_AUTOSIZE)

capture = cv.CaptureFromCAM(0)

sock = socket()
sock.connect(('192.168.0.2', 5001))
sock.send('Pi - Hallo')

while True:
    frame = cv.QueryFrame(capture)
    cv.ShowImage("camera", frame)

    mat = cv.GetMat(frame)
    buf = [1,90]

    image = cv.CreateImage (cv.GetSize (frame), 8, 3)
    nuImage = numpy.asarray(frame[:,:])
    imgencode = cv2.imencode('.png', nuImage, buf)
    data = numpy.array(imgencode)
    stringData = data.tostring()
    sock.send('Pi - Sending image data');
    sock.send( stringData );

    if cv.WaitKey(10) == 27:
        break

sock.send('Pi - closing connection')
sock.close()

1 个答案:

答案 0 :(得分:11)

您收到的数据不是垃圾。这是发送数据。但它是以字节发送的。因此,在接收端,您需要在处理数据之前将字节转换为字符串。阅读this以更好地理解在套接字编程中发送和接收数据,并阅读this以获取当前版本的opencv中使用的方法。

此外,你不需要使用cv来完成这项任务。 opencv的当前版本提供了仅由import cv2执行此操作的方法。

以下是工作代码,虽然简化了通信。

<强> client.py

#!/usr/bin/python
import socket
import cv2
import numpy

TCP_IP = 'localhost'
TCP_PORT = 5001

sock = socket.socket()
sock.connect((TCP_IP, TCP_PORT))

capture = cv2.VideoCapture(0)
ret, frame = capture.read()

encode_param=[int(cv2.IMWRITE_JPEG_QUALITY),90]
result, imgencode = cv2.imencode('.jpg', frame, encode_param)
data = numpy.array(imgencode)
stringData = data.tostring()

sock.send( str(len(stringData)).ljust(16));
sock.send( stringData );
sock.close()

decimg=cv2.imdecode(data,1)
cv2.imshow('CLIENT',decimg)
cv2.waitKey(0)
cv2.destroyAllWindows() 

<强> server.py

#!/usr/bin/python
import socket
import cv2
import numpy

def recvall(sock, count):
    buf = b''
    while count:
        newbuf = sock.recv(count)
        if not newbuf: return None
        buf += newbuf
        count -= len(newbuf)
    return buf

TCP_IP = 'localhost'
TCP_PORT = 5001

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind((TCP_IP, TCP_PORT))
s.listen(True)
conn, addr = s.accept()

length = recvall(conn,16)
stringData = recvall(conn, int(length))
data = numpy.fromstring(stringData, dtype='uint8')
s.close()

decimg=cv2.imdecode(data,1)
cv2.imshow('SERVER',decimg)
cv2.waitKey(0)
cv2.destroyAllWindows()