import sys import cv2 from showPic import Ui_MainWindow from PyQt5 import QtGui from PyQt5.QtCore import * from PyQt5.QtGui import * from PyQt5.QtWidgets import * class videoShow(QMainWindow,Ui_MainWindow): def __init__(self): super().__init__() self.setupUi(self) @pyqtSlot() def on_pushButton_record_clicked(self): camera_path = 0 # 0:自带摄像头 1:外接摄像头 "xxx.mp4" "rtsp://admin:pwd@192.168.2.10/cam/..." capture = cv2.VideoCapture(camera_path) # 初始化播放器 流媒体 fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V') # XVID/DIVX MPEG MJPG X264 video_writer = cv2.VideoWriter("image/myself.mp4", fourcc, 25, (960, 540)) # 存放路径、、帧率fps、尺寸(且保证下面的frame也是这个尺寸) while True: flag, frame = capture.read() if flag is False: continue frame = cv2.resize(frame, (960, 540)) video_writer.write(frame) self.display_image(frame, self.label) # 显示帧到标签 key = cv2.waitKey(25) if key == 27: video_writer.release() break @pyqtSlot() def on_pushButton_play_clicked(self): video_path = "image/myself.mp4" # 已经录制好的视频路径 capture = cv2.VideoCapture(video_path) # 初始化播放器 while True: flag, frame = capture.read() if flag is False: break self.display_image(frame, self.label) # 显示帧到标签 key = cv2.waitKey(25) if key == 27: break capture.release() def display_image(self, frame, label): pic_image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # 将BGR格式图像转换成RGB height, width = pic_image.shape[:2] pixMap = QImage(pic_image.data, width, height, width * 3, QImage.Format_RGB888) # 将RGB格式图像转换为八位图 pixMap = QPixmap.fromImage(pixMap) ratio = max(width / self.label.width(), height / self.label.height()) pixMap.setDevicePixelRatio(ratio) # 根据图片比例显示 self.label.setAlignment(Qt.AlignCenter) # 设置居中 self.label.setPixmap(pixMap) if __name__ == '__main__': app = QApplication(sys.argv) ui = videoShow() ui.show() sys.exit(app.exec_())在这样的代码基础上实现点击按钮停止就停止录制并保存视频
时间: 2024-02-14 14:30:10 浏览: 136
showpic.zip_MFC 显示JPG图片
要在点击按钮时停止录制并保存视频,您可以在`on_pushButton_record_clicked`函数中添加一个停止录制的条件,并在满足条件时释放视频写入器。以下是修改后的代码示例:
```python
import sys
import cv2
from showPic import Ui_MainWindow
from PyQt5 import QtGui
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
class videoShow(QMainWindow,Ui_MainWindow):
def __init__(self):
super().__init__()
self.setupUi(self)
self.recording = False
@pyqtSlot()
def on_pushButton_record_clicked(self):
camera_path = 0 # 0:自带摄像头 1:外接摄像头 "xxx.mp4" "rtsp://admin:pwd@192.168.2.10/cam/..."
capture = cv2.VideoCapture(camera_path) # 初始化播放器 流媒体
fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V') # XVID/DIVX MPEG MJPG X264
video_writer = cv2.VideoWriter("image/myself.mp4", fourcc, 25, (960, 540)) # 存放路径、、帧率fps、尺寸(且保证下面的frame也是这个尺寸)
self.recording = True
while self.recording:
flag, frame = capture.read()
if flag is False:
continue
frame = cv2.resize(frame, (960, 540))
video_writer.write(frame)
self.display_image(frame, self.label) # 显示帧到标签
key = cv2.waitKey(25)
if key == 27:
break
video_writer.release()
@pyqtSlot()
def on_pushButton_play_clicked(self):
video_path = "image/myself.mp4" # 已经录制好的视频路径
capture = cv2.VideoCapture(video_path) # 初始化播放器
while True:
flag, frame = capture.read()
if flag is False:
break
self.display_image(frame, self.label) # 显示帧到标签
key = cv2.waitKey(25)
if key == 27:
break
capture.release()
@pyqtSlot()
def on_pushButton_stop_clicked(self):
self.recording = False
def display_image(self, frame, label):
pic_image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # 将BGR格式图像转换成RGB
height, width = pic_image.shape[:2]
pixMap = QImage(pic_image.data, width, height, width * 3, QImage.Format_RGB888) # 将RGB格式图像转换为八位图
pixMap = QPixmap.fromImage(pixMap)
ratio = max(width / self.label.width(), height / self.label.height())
pixMap.setDevicePixelRatio(ratio) # 根据图片比例显示
self.label.setAlignment(Qt.AlignCenter) # 设置居中
self.label.setPixmap(pixMap)
if __name__ == '__main__':
app = QApplication(sys.argv)
ui = videoShow()
ui.show()
sys.exit(app.exec_())
```
在修改后的代码中,我们添加了一个名为`recording`的属性来跟踪录制状态。在`on_pushButton_record_clicked`函数中,我们在循环中检查`self.recording`的值,如果它为False,则停止录制并释放视频写入器。我们还添加了一个名为`on_pushButton_stop_clicked`的槽函数,用于在点击停止按钮时将`self.recording`设置为False,从而停止录制。请确保在UI文件中存在一个名为`pushButton_stop`的按钮,并将其与`on_pushButton_stop_clicked`槽函数关联。
阅读全文