电脑摄像头拍照demo-python

python3代码,使用PYQT5的QCameraInfo得到摄像头列表,然后CV2库从摄像头得到图像并保存.

下载资源:https://download.csdn.net/download/zhoury/12328554

代码如下: 

#window_video.py
import sys
import os
import cv2
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtMultimedia import QCameraInfo

class Video(QMainWindow):
    def __init__(self):
        super().__init__()

        self.camid = -1      #当前摄像头编号
        self.initUi()
        self.get_cam()
        self._timer = QTimer(self)
        self._timer.timeout.connect(self.play)

    def initUi(self):
        self.centralwidget = QWidget()
        self.setCentralWidget(self.centralwidget)
        self.startButton = QPushButton("开启")
        self.closeButton = QPushButton("关闭")
        self.camButton = QPushButton("拍照")
        self.promptLabel = QLabel('请选择摄像头：', self)
        self.combo = QComboBox(self)    #摄像头列表
        hbox = QHBoxLayout()
        hbox.addWidget(self.promptLabel)
        hbox.addWidget(self.combo)
        hbox.addStretch(1)
        hbox.addWidget(self.startButton)
        hbox.addWidget(self.closeButton)
        hbox.addWidget(self.camButton)

        vbox = QVBoxLayout()
        vbox.addLayout(hbox)
        self.vF = QLabel()
        vbox.addWidget(self.vF)
        self.centralwidget.setLayout(vbox)

        self.setGeometry(300, 300, 300, 150)
        self.setWindowTitle('简单拍照')    

        self.startButton.clicked.connect(self.onStartVideo)
        self.closeButton.clicked.connect(self.onCloseVideo)
        self.camButton.clicked.connect(self.onCamera)
        self.combo.currentIndexChanged.connect(self.selectionchange)

        self.startButton.setEnabled(False)
        self.closeButton.setEnabled(False)
        self.camButton.setEnabled(False)

    def onStartVideo(self):
        
        if self.camid < 0:
            return

        # 初始化传入的摄像头句柄为实例变量,并得到摄像头宽度和高度
        # cv2.CAP_DSHOW
        self.cam = cv2.VideoCapture(self.camid,cv2.CAP_DSHOW)
        self.w = self.cam.get(cv2.CAP_PROP_FRAME_WIDTH)
        self.h = self.cam.get(cv2.CAP_PROP_FRAME_HEIGHT)

        # 设置GUI窗口的位置和尺寸
        self.setGeometry(300, 200, self.w+20, self.h+100)
        # 开启定时器 
        self._timer.start(50)

        self.startButton.setEnabled(False)
        self.closeButton.setEnabled(True)
        self.camButton.setEnabled(True)
        self.combo.setEnabled(False)

    def onCloseVideo(self):
        self._timer.stop()

        if self.camid >= 0:
            self.startButton.setEnabled(True)
            self.closeButton.setEnabled(False)
            self.camButton.setEnabled(False)
            self.combo.setEnabled(True)

    def selectionchange(self):
        self.camid = self.combo.currentIndex()
        self.startButton.setEnabled(True)

    def play(self):
        """
        从摄像头得到图像 先转换为RGB格式 再生成QImage对象
        再用此QImage刷新vF实例变量 以刷新视频画面
        """
        r, f = self.cam.read()
        if r:
            self.vF.setPixmap(QPixmap.fromImage(QImage(cv2.cvtColor(f, cv2.COLOR_BGR2RGB), self.w,self.h,13)))
            #cv2.imwrite("c:\\temp\\aaa.jpg",f, [int(cv2.IMWRITE_JPEG_QUALITY),95])

    def get_cam(self):    #使用QCameraInfo得到摄像头列表
        camlist = QCameraInfo.availableCameras()
        self.cnt = 0
        for cam in camlist:
            self.combo.addItem(str(self.cnt) + ' - ' + cam.description())
            self.cnt = self.cnt + 1
        if self.cnt == 0:
            ret = QMessageBox.information(self,"Attention","没有找到摄像头", QMessageBox.Ok )

    def onCamera(self):  #拍照并保存
        r, f = self.cam.read()
        if r:
            #self.vF.setPixmap(QPixmap.fromImage(QImage(cv2.cvtColor(f, cv2.COLOR_BGR2RGB), self.w,self.h,13)))
            #cv2.imwrite("c:\\temp\\aaa.jpg",f, [int(cv2.IMWRITE_JPEG_QUALITY),95])
            newfileName, ok = QFileDialog.getSaveFileName(self, "文件另存为", "", "*.jpg")
            if newfileName:
                cv2.imwrite(newfileName,f, [int(cv2.IMWRITE_JPEG_QUALITY),95])
                os.system(newfileName)

if __name__ == '__main__':
    app = QApplication(sys.argv)
    # 初始化GUI窗口 并传入摄像头句柄
    win = Video()
    win.show()
    sys.exit(app.exec_())

运行截图如下: