（17）Python读取摄像头并实现视频播放、暂停、指定目录保存、回放功能

读取摄像头并播放、暂停功能

1. import sys
2. #import scipy.io as sio
3. from PyQt5 import QtGui, QtCore, QtWidgets
4. #from wyc import Ui_Form
5. import cv2
6. import numpy as np
8. class VideoCapture(QtWidgets.QWidget):
9. def __init__(self, filename, parent):
10. super(QtWidgets.QWidget, self).__init__()
11. self.cap = cv2.VideoCapture(0)
12. self.video_frame = QtWidgets.QLabel()
13. parent.layout.addWidget(self.video_frame)
15. def nextFrameSlot(self):
16. ret, frame = self.cap.read()
17. # frame = cv2.cvtColor(frame, cv2.cv.CV_BGR2RGB)
19. frame2 = np.zeros((frame.shape), dtype=np.int8)
20. # BGR
21. frame2[:, :, 0] = frame[:, :, 2]
22. frame2[:, :, 1] = frame[:, :, 1]
23. frame2[:, :, 2] = frame[:, :, 0]
26. img = QtGui.QImage(frame2, frame2.shape[1], frame2.shape[0], QtGui.QImage.Format_RGB888)
27. pix = QtGui.QPixmap.fromImage(img)
28. self.video_frame.setPixmap(pix)
30. def start(self):
31. self.timer = QtCore.QTimer()
32. self.timer.timeout.connect(self.nextFrameSlot)
33. self.timer.start(1000.0/30)
35. def pause(self):
36. self.timer.stop()
38. def deleteLater(self):
39. self.cap.release()
40. super(QtGui.QWidget, self).deleteLater()
43. class VideoDisplayWidget(QtWidgets.QWidget):
44. def __init__(self,parent):
45. super(VideoDisplayWidget, self).__init__(parent)
47. self.layout = QtWidgets.QFormLayout(self)
49. self.startButton = QtWidgets.QPushButton('播放', parent)
50. self.startButton.clicked.connect(parent.startCapture)
51. self.startButton.setFixedWidth(50)
52. self.pauseButton = QtWidgets.QPushButton('暂停', parent)
53. self.pauseButton.setFixedWidth(50)
54. self.layout.addRow(self.startButton, self.pauseButton)
56. self.setLayout(self.layout)
59. class ControlWindow(QtWidgets.QMainWindow):
60. def __init__(self):
61. super(ControlWindow, self).__init__()
62. self.setGeometry(100, 100, 800, 600)
63. self.setWindowTitle("视频显示demo")
65. self.capture = None
67. self.matPosFileName = None
68. self.videoFileName = None
69. self.positionData = None
70. self.updatedPositionData = {'red_x':[], 'red_y':[], 'green_x':[], 'green_y': [], 'distance': []}
71. self.updatedMatPosFileName = None
73. self.isVideoFileLoaded = True
74. self.isPositionFileLoaded = True
76. self.quitAction = QtWidgets.QAction("&Exit", self)
77. self.quitAction.setShortcut("Ctrl+Q")
78. self.quitAction.setStatusTip('Close The App')
79. self.quitAction.triggered.connect(self.closeApplication)
81. # self.openMatFile = QtWidgets.QAction("&Open Position File", self)
82. # self.openMatFile.setShortcut("Ctrl+Shift+T")
83. # self.openMatFile.setStatusTip('Open .mat File')
84. # self.openMatFile.triggered.connect(self.loadPosMatFile)
86. # self.openVideoFile = QtWidgets.QAction("&Open Video File", self)
87. # self.openVideoFile.setShortcut("Ctrl+Shift+V")
88. # self.openVideoFile.setStatusTip('Open .h264 File')
89. # self.openVideoFile.triggered.connect(self.loadVideoFile)
91. # self.mainMenu = self.menuBar()
92. # self.fileMenu = self.mainMenu.addMenu('&File')
93. # self.fileMenu.addAction(self.openMatFile)
94. # self.fileMenu.addAction(self.openVideoFile)
95. # self.fileMenu.addAction(self.quitAction)
97. self.videoDisplayWidget = VideoDisplayWidget(self)
98. self.setCentralWidget(self.videoDisplayWidget)
100. def startCapture(self):
101. if not self.capture and self.isPositionFileLoaded and self.isVideoFileLoaded:
102. self.capture = VideoCapture(self.videoFileName, self.videoDisplayWidget)
103. self.videoDisplayWidget.pauseButton.clicked.connect(self.capture.pause)
104. self.capture.start()
106. def endCapture(self):
107. self.capture.deleteLater()
108. self.capture = None
110. # def loadPosMatFile(self):
111. # try:
112. # self.matPosFileName = str(QtGui.QFileDialog.getOpenFileName(self, 'Select .mat position File'))
113. # self.positionData = sio.loadmat(self.matPosFileName)
114. # self.isPositionFileLoaded = True
115. # except:
116. # print("Please select a .mat file")
118. # def loadVideoFile(self):
119. # try:
120. # self.videoFileName = QtGui.QFileDialog.getOpenFileName(self, 'Select .h264 Video File')
121. # self.isVideoFileLoaded = True
122. # except:
123. # print("Please select a .h264 file")
125. def closeApplication(self):
126. choice = QtGui.QMessageBox.question(self, 'Message','Do you really want to exit?',QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
127. if choice == QtGui.QMessageBox.Yes:
128. print("Closing....")
129. sys.exit()
130. else:
131. pass
134. if __name__ == '__main__':
135. import sys
136. app = QtWidgets.QApplication(sys.argv)
137. window = ControlWindow()
138. window.show()
139. sys.exit(app.exec_())

回放、指定目录保存功能

1. 以时间戳的形式保存视频：
2. import cv2
3. import datetime
5. cap = cv2.VideoCapture(0)
6. import os.path
7. save_path = 'G:/Temp/' #以特定前缀保存视频文件到指定目录
8. timeNow = "%s.avi" % (datetime.datetime.now().strftime('%Y_%m_%d_%H-%M-0%S'))
9. completeName = os.path.join(save_path, timeNow)
11. out = (cv2.VideoWriter(timeNow, cv2.VideoWriter_fourcc(*'PIM1'), 30.0, (640, 480))) #第三个参数是帧率
12. while cap.isOpened():
13. ret, frame = cap.read()
14. if ret==True:
15. out.write(frame)
16. cv2.imshow('frame', frame)
17. if cv2.waitKey(1) & 0xFF == ord('q'):
18. break
19. else:
20. break
22. # Release everything if job is finished
23. cap.release()
24. out.release()
25. cv2.destroyAllWindows()
28. 回放视频：
29. import os
30. def videoBack(self):
31. os.startfile("G:/Tx2Project/VideoShow") #录制的视频存放文件