Socket传图后RAM中显示遇OpenCV报错，求解决方案及数组优化

一、解决imshow报错问题 #

你遇到的cv2.error: (-215:Assertion failed)错误，核心原因是处理后的数组arr2 dtype为32位整数（int32），而OpenCV的imshow要求输入图像必须是8位无符号整数（uint8）——cv2.imread读取的原始图像img默认就是uint8类型，你用np.fromstring(imgstr, dtype=int, sep=" ")时，int默认对应int32，导致类型不匹配触发断言失败。

修复代码 #

在调用imshow前，将数组强制转换为uint8类型：

# 转换数组类型为uint8
arr2 = arr2.astype(np.uint8)
# 显示图像（需补充waitKey，否则窗口会直接关闭）
cv2.imshow("hi", arr2)
cv2.waitKey(0)
cv2.destroyAllWindows()

二、数组处理速度优化 #

你当前的嵌套循环、字符串转换、多次vstack操作完全冗余，效率极低。无论是本地模拟处理还是Socket传输图像，都应该直接操作numpy数组的字节数据，避免无意义的格式转换。

优化后的本地模拟处理（替代你当前的冗余逻辑） #

如果只是模拟“从字节流还原图像”的过程，直接用numpy的字节转换即可：

import cv2
import numpy as np

# 读取原始图像
img = cv2.imread("resim/tfm.jpg")
# 获取图像的形状和原始类型
img_shape = img.shape
img_dtype = img.dtype

# 模拟Socket发送：数组转字节流
img_bytes = img.tobytes()

# 模拟Socket接收：字节流转回数组
arr2 = np.frombuffer(img_bytes, dtype=img_dtype).reshape(img_shape)

# 验证一致性
print(arr2 == img)

# 直接显示（类型已匹配，无需额外转换）
cv2.imshow("hi", arr2)
cv2.waitKey(0)
cv2.destroyAllWindows()

实际Socket传输的完整示例 #

服务器端（发送图像）

import cv2
import numpy as np
import socket

# 读取图像
img = cv2.imread("resim/tfm.jpg")
shape = img.shape
dtype = img.dtype

# 创建Socket服务
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind(('localhost', 12345))
server_socket.listen(1)

print("等待客户端连接...")
conn, addr = server_socket.accept()
print(f"连接来自: {addr}")

# 先发送图像形状（3个int32，共12字节）
conn.sendall(np.array(shape, dtype=np.int32).tobytes())
# 发送图像类型编码（方便客户端还原dtype）
conn.sendall(np.array([np.dtype(dtype).num], dtype=np.int32).tobytes())
# 发送图像字节数据
conn.sendall(img.tobytes())

conn.close()
server_socket.close()

客户端（接收并内存显示）

import cv2
import numpy as np
import socket

# 连接服务器
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client_socket.connect(('localhost', 12345))

# 接收图像形状
shape_data = client_socket.recv(12)
shape = np.frombuffer(shape_data, dtype=np.int32)

# 接收图像类型编码
dtype_num_data = client_socket.recv(4)
dtype_num = np.frombuffer(dtype_num_data, dtype=np.int32)[0]
dtype = np.dtype(dtype_num)

# 接收完整图像字节数据
total_bytes = np.prod(shape) * dtype.itemsize
img_bytes = b''
while len(img_bytes) < total_bytes:
    chunk = client_socket.recv(min(4096, total_bytes - len(img_bytes)))
    if not chunk:
        break
    img_bytes += chunk

# 还原数组并显示
img = np.frombuffer(img_bytes, dtype=dtype).reshape(shape)
cv2.imshow("Received Image", img)
cv2.waitKey(0)
cv2.destroyAllWindows()

client_socket.close()

内容的提问来源于stack exchange，提问作者saitamatechno