再添加2个按钮,一个按钮负责暂停摄像头 一个负责开始摄像头
import sys
import cv2
from PyQt5.QtCore import QTimer, Qt
from PyQt5.QtGui import QImage, QPixmap
from PyQt5.QtWidgets import QApplication, QLabel, QVBoxLayout, QWidget, QPushButton
class CameraWidget(QWidget):
def __init__(self):
super().__init__()
self.camera = cv2.VideoCapture(0) # 打开摄像头
self.image_label = QLabel(self) # 显示图像的Label
layout = QVBoxLayout(self)
layout.addWidget(self.image_label)
self.timer = QTimer() # 定时器,用于不断读取图像
self.timer.timeout.connect(self.update_frame)
self.start_button = QPushButton('开始摄像头', self) # 开始摄像头按钮
self.start_button.clicked.connect(self.start_camera)
layout.addWidget(self.start_button)
self.pause_button = QPushButton('暂停摄像头', self) # 暂停摄像头按钮
self.pause_button.clicked.connect(self.pause_camera)
layout.addWidget(self.pause_button)
self.start_camera() # 默认开始摄像头
self.resize(800, 600) # 设置窗口初始大小
def start_camera(self):
if not self.timer.isActive():
self.timer.start(30) # 启动定时器
def pause_camera(self):
if self.timer.isActive():
self.timer.stop() # 停止定时器
def update_frame(self):
ret, frame = self.camera.read() # 读取摄像头图像
if ret:
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # 转换颜色空间
image = QImage(frame, frame.shape[1], frame.shape[0], QImage.Format_RGB888)
pixmap = QPixmap.fromImage(image)
self.image_label.setPixmap(pixmap.scaled(self.image_label.size().width() - 20, self.image_label.size().height() - 20, Qt.KeepAspectRatio))
if __name__ == "__main__":
app = QApplication(sys.argv)
widget = CameraWidget()
widget.show()
sys.exit(app.exec_())
在上述代码中,我们添加了start_button
和pause_button
两个按钮,并将它们分别连接到了start_camera
和pause_camera
方法。在start_camera
方法中,如果定时器未激活,则启动定时器开始读取摄像头图像;在pause_camera
方法中,如果定时器处于激活状态,则停止定时器暂停读取摄像头图像。
这样,您就可以通过点击按钮来控制摄像头的开始和暂停。默认情况下,程序一开始就会自动开始摄像头。您可以根据需要自由调整按钮文本和连接的方法。