電腦攝像頭拍照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_())

運行截圖如下: