一个基于socket的高速视频传输程序

前天才申请的这个博客看来访问量不错,本来写这个博客的原因之一就是好好学习python,所以我会经常写一些东西的.希望大家多多光临.

需要说明的是,本blog上的文章除了我特别说明摘抄的外,都是个人的作品.

今天要说的是一个高速视频流的采集和传输的问题,我不是研究这一块的,没有使用什么算法,仅仅是兴趣导致我很想搞懂这个问题.

1,首先是视频数据[摄像头图像]的采集,通常可以使用vfw在vc或者vb下实现,这个库我用的不好,所以一直不怎么会用.现在我们用到的是python的videocapture库,这个库用起来很简单,如下:

from VideoCapture import Device cam = Device() cam.setResolution(320,240) #设置显示分辨率 cam.saveSnapshot('demo.jpg') #抓取并保存图片

这样,你就得到了一张图片.

当然,要实现比较高速的采集,每次都要保存图片是不大可能的,这样弄下来每次采集一张图片已经接近1秒钟,这个速度我们是无法忍受的.

更好的解决方式是直接这样: im = cam.getImage(),返回的是一个Image对象,是一个内存块,对它操作就快速多了.

2,如何传输图片?我不知道pplive这种视频是怎么传输的,我的想法很简单,每次传送一幅图片.

在本程序中,每次传送的RGB图像的大小为 160*120.这样,需要的数据量为:

d = 160*120*3 = 56.25 kB

我选择80000B.

下面是发送端代码:

import socket import Image from VideoCapture import Device cam = Device() cam.setResolution(320,240) clisocket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) while 1: im = cam.getImage() im = im.resize((160,120)) da = im.tostring() clisocket.sendto(da, ("127.0.0.1", 1234)) s.close()

3,如何实时显示图片?

我使用了pygame作为实时图像显示界面,因为pygame是经过优化的高速图形库,不知道有没有使用directshow,我想应该用了吧..

关于pygame请参考 www.pygame
.org

下面是接收端代码:

import socket import Image import os,sys,pygame from pygame.locals import * pygame.init() screen = pygame.display.set_mode((160,120)) pygame.display.set_caption("web cam") pygame.display.flip() svrsocket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) svrsocket.bind(("127.0.0.1", 1234)) clock = pygame.time.Clock() #计算帧速 while 1: data, address = svrsocket.recvfrom(80000) camshot = pygame.image.frombuffer(data, (160,120), "RGB") for event in pygame.event.get(): if event.type == pygame.QUIT: sys.exit() screen.blit(camshot, (0,0)) pygame.display.update() print clock.get_fps() #在终端打印帧速 clock.tick()

程序终于完成了,测试一下效果怎么样把,为了方便,我把客户端和服务端都设成了本机,端口为1234.

运行程序,哇,简直不可思议,最高竟然达到230fps!(右边的终端)