用代码编写使用socket用自定义的数据帧传输视频流

以下是一个使用 socket 自定义数据帧传输视频流的示例代码：

import cv2
import numpy as np
import struct
import socket

# 定义视频编解码器
fourcc = cv2.VideoWriter_fourcc(*'XVID')
# 打开本地摄像头
cap = cv2.VideoCapture(0)
# 设置视频帧大小
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)

# 定义数据帧头部
HEADERSIZE = 4
# 定义服务器 IP 和端口号
SERVER_IP = 'localhost'
SERVER_PORT = 1234

# 创建 socket 对象
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((SERVER_IP, SERVER_PORT))

# 不断读取视频帧并传输
while True:
    # 读取视频帧
    ret, frame = cap.read()

    # 编码视频帧
    _, img_encoded = cv2.imencode('.jpg', frame, [cv2.IMWRITE_JPEG_QUALITY, 90])

    # 构造数据帧
    msg = struct.pack('>I', len(img_encoded)) + img_encoded
    # 发送数据帧
    s.sendall(msg)

    # 接收服务器确认信息
    data = s.recv(1024)
    print(data.decode())

# 关闭 socket 对象和摄像头
s.close()
cap.release()

在服务器端，我们可以使用以下代码来接收数据帧并显示视频流：

import cv2
import numpy as np
import struct
import socket

# 定义视频编解码器
fourcc = cv2.VideoWriter_fourcc(*'XVID')
# 创建 VideoWriter 对象
out = cv2.VideoWriter('output.avi', fourcc, 20.0, (640, 480))

# 定义数据帧头部
HEADERSIZE = 4
# 定义服务器 IP 和端口号
SERVER_IP = 'localhost'
SERVER_PORT = 1234

# 创建 socket 对象并绑定端口号
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind((SERVER_IP, SERVER_PORT))
s.listen(1)

# 等待客户端连接
conn, addr = s.accept()
print('Connected by', addr)

# 不断接收数据帧并显示
while True:
    # 接收数据帧头部
    data = conn.recv(HEADERSIZE)
    # 如果头部为空，说明客户端已经关闭连接
    if not data:
        break
    # 解析数据帧头部，获取数据帧大小
    msglen = struct.unpack('>I', data)[0]
    # 接收数据帧
    data = conn.recv(msglen)
    # 将数据帧转换成图像
    img = cv2.imdecode(np.frombuffer(data, np.uint8), cv2.IMREAD_COLOR)
    # 显示图像
    cv2.imshow('frame', img)
    # 写入视频文件
    out.write(img)

    # 发送确认信息
    conn.sendall(b'OK')

    # 按下 q 键退出循环
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 关闭 socket 对象和视频文件
conn.close()
s.close()
out.release()
cv2.destroyAllWindows()

注意：这只是一个简单的示例代码，实际应用中需要考虑数据的丢失和重传，以及帧率、码率等问题。