python opencv 通過Soket傳輸圖像

最近需要將本地的實時視頻圖像傳輸到遠端，故採用Soket TCP傳輸實現。其中也遇到了很多問題，最終解決。

服務端：

import cv2
import time
import numpy as np
import socket


def recv_size(sock, count):

    buf = b''#buf是一個byte類型

    while count:

        newbuf = sock.recv(count)

        if not newbuf: 

            return None

        buf += newbuf

        count -= len(newbuf)

    return buf



def recv_img(sock, count):

    buf = b''#buf是一個byte類型

    while count:

        #接受TCP套接字的數據。數據以字符串形式返回，count指定要接收的最大數據量.

        newbuf = sock.recv(count)

        if not newbuf: 

            return None

        buf += newbuf

        count -= len(newbuf)

    return buf



# socket.AF_INET 用於服務器與服務器之間的網絡通信

# socket.SOCK_STREAM 代表基於TCP的流式socket通信

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)



# sock.bind(('127.0.0.1', 00000))

sock.bind(('0.0.0.0', 00000))  #需改爲自己的ip和端口

sock.listen(3)



conn, addr = sock.accept()

sentace = "get_size"



def get_img():

    start = time.time()

    length = recv_size(conn, 16)



    conn.send(sentace.encode('utf-8'))



    result = int(str(length, encoding='utf-8'))



    stringData = recv_img(conn, result)



    data = np.fromstring(stringData, np.uint8)

    decimg = cv2.imdecode(data, cv2.IMREAD_COLOR)  # 解碼處理，返回mat圖片



    end = time.time()

    seconds = end - start

    print("Time is: " + str(seconds))



    return decimg



def send_result(img):

    ret, img_encode = cv2.imencode('.jpg', img)

    data_encode = np.array(img_encode)



    stringData = data_encode.tostring()



    # 首先發送圖片編碼後的長度

    conn.send(str.encode(str(len(stringData)).ljust(16)))

    answer = conn.recv(1024)

    answer =  str(answer, encoding = "utf-8")

    if answer == "Get":

        conn.send(stringData)

客戶端：

import cv2
import socket
import sys
import numpy as np
import time

cap = cv2.VideoCapture(0)

def load_img():
    _, frame = cap.read()
    image = frame[..., ::-1]
    # image  = frame
    return image

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 連接服務端
sock.connect(('10.22.18.23', 52846))

# socket.setdefaulttimeout(1000)


def recv_img(sock, count):
    buf = b''#buf是一個byte類型
    while count:
        #接受TCP套接字的數據。數據以字符串形式返回，count指定要接收的最大數據量.
        newbuf = sock.recv(count)
        if not newbuf: 
            return None
        buf += newbuf
        count -= len(newbuf)
    return buf

encode_param = [int(cv2.IMWRITE_JPEG_QUALITY), 100]
sentace = "Get"
while True:
    img = load_img()

    # 首先對圖片進行編碼，因爲socket不支持直接發送圖片
    # '.jpg'表示把當前圖片frame按照jpg格式編碼
    ret, img_encode = cv2.imencode('.jpg', img, encode_param)
    data_encode = np.array(img_encode)

    stringData = data_encode.tostring()

    # 首先發送圖片編碼後的長度
    sock.send(str.encode(str(len(stringData)).ljust(16)))

    answer = sock.recv(1024)
    answer =  str(answer, encoding = "utf-8")
    if answer == "get_size":
        sock.send(stringData)
    
    # 接收返回的數據尺寸
    receive = sock.recv(1024)

    # 返回信息已收到通知
    sock.send(sentace.encode('utf-8'))
    receive = int(str(receive, encoding='utf-8'))
    if receive:
        result_mat = recv_img(sock, receive)
        result_mat = np.fromstring(result_mat, np.uint8)
        result_mat = cv2.imdecode(result_mat, cv2.IMREAD_COLOR)  # 解碼處理，返回mat圖片
        cv2.imshow("Backgrpound Matting", result_mat)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

sock.close()
cv2.destroyAllWindows()