基于opencv和numpy实现图像文件的传输

引言

网上大部分文件的传输都是基于os库，将文件打开为字节而后逐字节发送的。

而我在实践过程中尝试了一种新的方式，效果尚可。特意拿来分享。

实现原理

客户端将图像打开为矩阵并转化为一维，而后将这个一维数字转化为字节数组。第一帧发送图像的形状，之后将这个字节数组分组发送出去，每次最多500字节，直到发送完毕后，发送一个end。服务器收到图像后，先将其存储到一个一维数组里，全部接收完毕后，在按所给形状重新reshape。

传图像代码

客户端

import time

import cv2
import socket
import numpy as ny

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
addr = ('127.0.0.1', 12345)

img = cv2.imread('qq.jpeg')
shape = img.shape  # 记录图像形状
str_shape = ' '.join([str(s) for s in shape])  # 形状转化为便于发送的字符传
s.sendto(str_shape.encode('utf-8'), addr)
img_fl = img.flatten()  # 将图像变为一维
img_bytes = bytes(img_fl)
# 将图像变为bytes
cut = lambda obj, sec: [obj[i:i + sec] for i in range(0, len(obj), sec)]
# 按指定大小拆分
byteList = cut(img_bytes, 500)
for by in byteList:
    s.sendto(by, addr)
    time.sleep(0.001)
time.sleep(0.01)
s.sendto(b'end', addr)

服务器

import socket
import numpy as np
import cv2

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.bind(('127.0.0.1', 12345))
data, _ = s.recvfrom(1024)
shape = [int(s) for s in data.decode('utf-8').split() if len(s) > 0