socket 分包传输mat示例代码

以下是一个简单的示例代码，展示了如何使用 socket 分包传输 mat：

发送方代码：

import socket
import numpy as np
import cv2

# 将 mat 转换为字节数组
def mat_to_bytes(mat):
    return mat.tobytes()

# 创建 socket 对象
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 连接服务器
server_address = ('localhost', 8888)
client_socket.connect(server_address)

# 加载图像
img = cv2.imread('test.jpg')

# 将图像转换为 mat 对象
mat = np.array(img)

# 将 mat 转换为字节数组
data = mat_to_bytes(mat)

# 发送包头
header = len(data).to_bytes(4, byteorder='big')
client_socket.sendall(header)

# 分包发送数据
buffer_size = 1024
offset = 0
while offset < len(data):
    size = min(buffer_size, len(data) - offset)
    client_socket.sendall(data[offset:offset+size])
    offset += size

# 关闭连接
client_socket.close()

接收方代码：

import socket
import numpy as np
import cv2

# 将字节数组转换为 mat
def bytes_to_mat(data):
    mat = np.frombuffer(data, dtype=np.uint8)
    mat = np.reshape(mat, (-1, 3))
    mat = np.reshape(mat, (height, width, 3))
    return mat

# 创建 socket 对象
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 绑定地址和端口号
server_address = ('localhost', 8888)
server_socket.bind(server_address)

# 开始监听
server_socket.listen(1)

# 等待客户端连接
client_socket, client_address = server_socket.accept()

# 接收包头
header = client_socket.recv(4)
data_size = int.from_bytes(header, byteorder='big')

# 接收数据
buffer_size = 1024
data = b''
while len(data) < data_size:
    recv_size = min(buffer_size, data_size - len(data))
    recv_data = client_socket.recv(recv_size)
    data += recv_data

# 将字节数组转换为 mat 对象
img = bytes_to_mat(data)

# 显示图像
cv2.imshow('image', img)
cv2.waitKey(0)

# 关闭连接
client_socket.close()
server_socket.close()

请注意，这只是一个简单的示例代码，实际应用中还需要考虑数据的完整性和正确性，以及网络传输的稳定性等因素。