基于python实现H.264视频推流与接收

H.264是一种高效的视频编解码标准，常被用于实时视频传输。实现H.264视频推流和接收可以使用Python中的OpenCV库和FFmpeg工具。

推流端：

1. 导入OpenCV库和FFmpeg工具，使用OpenCV库读取本地视频文件，打开视频流。

import cv2
import subprocess

# 打开视频文件
cap = cv2.VideoCapture('test_video.mp4')

# 获取视频帧率
fps = int(cap.get(cv2.CAP_PROP_FPS))

# 设置视频编码器为H.264
fourcc = cv2.VideoWriter_fourcc(*'H264')

# 创建输出视频流
output = cv2.VideoWriter('udp://127.0.0.1:1234', fourcc, fps, (640, 480))

2. 使用FFmpeg工具将视频流推送到目标地址。

# 启动FFmpeg进程
cmd = 'ffmpeg -i - -vcodec copy -f mpegts udp://127.0.0.1:1234'
p = subprocess.Popen(cmd.split(), stdin=subprocess.PIPE)

# 推流
while True:
    ret, frame = cap.read()
    if ret:
        output.write(frame)
        p.stdin.write(frame.tostring())
    else:
        break

# 关闭FFmpeg进程和视频流
p.stdin.close()
output.release()
cap.release()

接收端：

1. 导入OpenCV库和FFmpeg工具，使用FFmpeg工具接收视频流，并将其转换为OpenCV中的Mat格式。

import cv2
import subprocess

# 启动FFmpeg进程
cmd = 'ffmpeg -i udp://127.0.0.1:1234 -f rawvideo -pix_fmt bgr24 -'
p = subprocess.Popen(cmd.split(), stdout=subprocess.PIPE)

# 接收视频流
while True:
    # 读取视频帧
    raw_frame = p.stdout.read(640*480*3)
    if len(raw_frame) != 640*480*3:
        break

    # 转换为OpenCV中的Mat格式
    frame = np.frombuffer(raw_frame, np.uint8).reshape((480, 640, 3))

    # 显示视频帧
    cv2.imshow('frame', frame)
    cv2.waitKey(1)

# 关闭FFmpeg进程和窗口
p.stdout.close()
cv2.destroyAllWindows()

以上代码仅供参考，实际使用时需要根据具体情况进行修改。同时需要注意的是，H.264视频编解码需要较高的计算资源，推荐在较高配置的设备上进行测试。