PyQt+OpenCV 录制保存、播放视频
准备这一两个月看看以前的书，另外学习下视频方面的东西（CV方面）。
之前学过几天QT/PyQt，于是用PyQt结合OpenCV，做了一个小程序。具有播放并保存摄像头捕获组成的视频，以及播放本地视频文件两个小功能。
界面很简单，没图没真相 。
代码就100多行，很容易看懂。最后感叹下OpenCV和Python的强大，噢，还有QT/PyQt。。
#! /usr/bin/env python
#coding=utf-8
from PyQt4.QtCore import *
from PyQt4.QtGui import *
import sys
import opencv
from PIL import Image
#this is important for capturing/displaying images
from opencv import highgui
from opencv import cv
import time
MPEG1VIDEO = 0x314D4950
camera = highgui.cvCreateCameraCapture(0)#找摄像头，一般填0-99都可以
class MainWindow(QWidget):
def __init__(self, parent = None):
QWidget.__init__(self)
self.resize(550, 550)
self.setWindowTitle('vedio control')
self.status = 0 #0 is init status;1 is play video; 2 is capture video
self.image = QImage()
#录制的视频保存位置、格式等参数设定
self.videowriter = highgui.cvCreateVideoWriter("test.mpg", highgui.CV_FOURCC('m','p','g','1'), 25, cv.cvSize(200,200), 1)
#播放的视频位置
self.playcapture = highgui.cvCreateFileCapture("test.avi")
#初始化按钮
self.capturebtn = QPushButton('capture')
self.playbtn = QPushButton('play')
exitbtn = QPushButton('exit')
# 界面布局
vbox = QVBoxLayout()
vbox.addWidget(self.capturebtn)
vbox.addWidget(self.playbtn)
vbox.addWidget(exitbtn)
self.piclabel = QLabel('pic')
hbox = QHBoxLayout()
hbox.addLayout(vbox)
hbox.addStretch(1)
hbox.addWidget(self.piclabel)
self.setLayout(hbox)
#加载初始页面
if self.image.load("1.jpg"):
self.piclabel.setPixmap(QPixmap.fromImage(self.image))
#设定定时器
self.timer = Timer() #录制视频
self.playtimer = Timer("updatePlay()")#播放视频
#信号--槽
self.connect(self.timer, SIGNAL("updateTime()"),
self.CaptureVGA)
self.connect(self.capturebtn, SIGNAL("clicked()"),
self.PauseBegin)
self.connect(self.playtimer, SIGNAL("updatePlay()"),
self.PlayVideo)
self.connect(self.playbtn, SIGNAL("clicked()"),
self.VideoPlayPause)
self.connect(exitbtn, SIGNAL("clicked()"),
app, SLOT("quit()"))
def PlayVideo(self):
im = highgui.cvQueryFrame(self.playcapture)
im = opencv.adaptors.Ipl2PIL(im)
im = im.convert('RGB').tostring('jpeg', 'RGB')
self.image.loadFromData(QByteArray(im))
self.piclabel.setPixmap(QPixmap.fromImage(self.image))
def VideoPlayPause(self):
self.status, playstr, capturestr = ((1, 'pause', 'capture'), (0, 'play', 'capture'), (1, 'pause', 'capture'))[self.status]#三种状态分别对应的显示、处理
self.playbtn.setText(playstr)
self.capturebtn.setText(capturestr)
if self.status is 1:#状态1，播放视频
self.timer.stop()
self.playtimer.start()
else:
self.playtimer.stop()
def PauseBegin(self):
self.status, playstr, capturestr = ((2, 'play', 'pause'), (2, 'play', 'pause'), (0, 'play', 'capture'))[self.status]
self.capturebtn.setText(capturestr)
self.playbtn.setText(playstr)
print self.status, playstr, capturestr
if self.status is 2:#状态2，录制并显示视频
self.timer.start()
self.playtimer.stop()
else:
self.timer.stop()
def CaptureVGA(self):
im = highgui.cvQueryFrame(camera)
highgui.cvWriteFrame(self.videowriter, im)#录制视频，写入文件
#convert Ipl image to PIL image
im = opencv.adaptors.Ipl2PIL(im)
im = im.convert('RGB').tostring('jpeg', 'RGB')# 转换格式，jpeg
self.image.loadFromData(QByteArray(im))#格式支持QT，直接加载
#im.save('3.jpg')#opencv 返回的是Ipl 格式，QT无法直接显示。不知道如何转换格式，采用保存、读取的方式。
#pic.load('3.jpg')
self.piclabel.setPixmap(QPixmap.fromImage(self.image)) #一帧一帧的显示
class Timer(QThread):
def __init__(self, signal = "updateTime()", parent=None):
super(Timer, self).__init__(parent)
self.stoped = False
self.signal = signal
self.mutex = QMutex()
def run(self):
with QMutexLocker(self.mutex):
self.stoped = False
while True:
if self.stoped:
return
self.emit(SIGNAL(self.signal))
time.sleep(0.04) #40毫秒发送一次信号，每秒25帧
def stop(self):
with QMutexLocker(self.mutex):
self.stoped = True
def isStoped(self):
with QMutexLocker(sellf.mutex):
return self.stoped
if __name__ == "__main__" :
app = QApplication(sys.argv)
main = MainWindow()
main.show()
sys.exit(app.exec_())