主页 / 问题 / 1541012
Accepted
bluesky
Asked:2023-09-16 23:24:17 +0000 UTC2023-09-16 23:24:17 +0000 UTC

如何在 PyQt 中从相机获取图像数组？

  • 772

任务是从网络摄像头接收一系列图像，对其进行过滤，选择一种颜色（例如，仅绿色）并将其显示在窗口中。

使用OpenCV很容易做到这一点，但是使用PyQt5时有两个问题：

  1. 如何获取图像数组？
  2. 如何在窗口中显示修改后的图像？

所提供的代码显示来自网络摄像头的原始图像：

import sys        
from PyQt5.Qt import *


class MyWidget(QWidget):
    def __init__(self, parent):            
        super(QWidget, self).__init__(parent)

        self.layout = QVBoxLayout(self)            
        self.viewfinder = QCameraViewfinder()
        self.cameraInfo = QCameraInfo.defaultCamera()            
        self.camera = QCamera(self.cameraInfo)
        self.camera.setViewfinder(self.viewfinder)
        self.camera.setCaptureMode(QCamera.CaptureStillImage)
        self.layout.insertWidget(1, self.viewfinder)
        if self.camera.status() != QCamera.UnavailableStatus:
            self.camera.start()  
                        

class MainWindow(QMainWindow):
    def __init__(self, *args, **kwargs):
        super(MainWindow, self).__init__(*args, **kwargs)

        self.widget1 = MyWidget(self)
        self.central_widget = QWidget()
        self.setCentralWidget(self.central_widget)
        main_layout = QVBoxLayout(self.central_widget)
        main_layout.addWidget(self.widget1)


if __name__ == '__main__':
    app = QApplication(sys.argv)
    app.setApplicationName("pyqt5")
    window = MainWindow()
    window.show()
    sys.exit(app.exec())

更新：上面介绍了使用OpenCV
在图像中显示绿色的代码。

需要在PyQt5窗口中显示相同的图像。

    import cv2
    import numpy as np


    # Границы зеленого цвета в формате 'HLS'
    hls_min = np.array((30, 0, 0), np.uint8)
    hls_max = np.array((100, 255, 255), np.uint8)    
    cap = cv2.VideoCapture(0)
    while True:
        # Трёхмерный массив оригинального изображения в формате 'BGR'
        ret, frame = cap.read()
        # Перевод из 'bgr' в 'hls' (нужно именно в 'HLS')
        hls = cv2.cvtColor(frame, cv2.COLOR_BGR2HLS) # Трёхмерный массив изображения в формате 'HLS'
        # Фильтр
        img_mask = cv2.inRange(hls, hls_min, hls_max)
        # Применение фильтра цвета
        img_filt = cv2.bitwise_and(frame, frame, mask = img_mask)
        a=cv2.waitKey(1)
        a=cv2.imshow("Filtered image", img_filt)
python

1 个回答

  1. Best Answer

    我不确定我是否正确理解你想要做什么。

    您写道：“...您需要在 PyQt5 窗口中显示相同的图像...”

    或者它可能看起来像这样：

    import sys
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.Qt import *
import numpy as np
import cv2


class Ui_CameraPage(object):
    def setupUi(self, CameraPage):
        CameraPage.setObjectName("CameraPage")
        CameraPage.resize(640, 480)
        self.layoutWidget = QtWidgets.QWidget(CameraPage)
        self.layoutWidget.setObjectName("layoutWidget")
        self.gridLayout = QtWidgets.QGridLayout(CameraPage) 
        self.gridLayout.setContentsMargins(3, 3, 3, 3)
        self.gridLayout.setObjectName("gridLayout")
        self.cameraButton = QtWidgets.QPushButton(self.layoutWidget)
        font = QtGui.QFont()
        font.setPointSize(12)
        self.cameraButton.setFont(font)
        self.cameraButton.setObjectName("cameraButton")
        self.gridLayout.addWidget(self.cameraButton, 0, 0, 1, 1)
        self.cameraLabel = QtWidgets.QLabel(self.layoutWidget, alignment=Qt.AlignCenter)
        self.cameraLabel.setMinimumSize(QtCore.QSize(480, 320))
        font = QtGui.QFont()
        font.setPointSize(12)
        self.cameraLabel.setFont(font)
        self.cameraLabel.setObjectName("cameraLabel")
        self.gridLayout.addWidget(self.cameraLabel, 0, 1, 1, 1)
        self.gridLayout.setColumnStretch(0, 0)
        self.gridLayout.setColumnStretch(1, 1)

        self.retranslateUi(CameraPage)
        QtCore.QMetaObject.connectSlotsByName(CameraPage)

    def retranslateUi(self, CameraPage):
        _translate = QtCore.QCoreApplication.translate
        CameraPage.setWindowTitle(_translate("CameraPage", "Camera Interface"))
        self.cameraButton.setText(_translate("CameraPage", "Camera On"))
        self.cameraLabel.setText(_translate("CameraPage", "Camera "))
        

class CameraPageWindow(QWidget, Ui_CameraPage):
    def __init__(self):
        super().__init__()
        self.setupUi(self)
        
        # Границы зеленого цвета в формате 'HLS'
        self.hls_min = np.array((30, 0, 0), np.uint8)
        self.hls_max = np.array((100, 255, 255), np.uint8) 
        
        self.timer_camera = QTimer() 
        self.timer_camera.timeout.connect(self.show_camera)
        self.cameraButton.clicked.connect(self.slotCameraButton)

    def show_camera(self):
        flag, self.image = self.cap.read()
        
        if self.image is not None:
            show = cv2.cvtColor(self.image, cv2.COLOR_BGR2HLS)
            # Фильтр
            img_mask = cv2.inRange(show, self.hls_min, self.hls_max)
            img_filt = cv2.bitwise_and(self.image, self.image, mask=img_mask)
            showImage = QImage(
                img_filt.data, 
                img_filt.shape[1], img_filt.shape[0],
                QImage.Format_RGB888
            )
            self.cameraLabel.setPixmap(QPixmap.fromImage(showImage))
        else:
            self.timer_camera.stop()
            self.cameraButton.setText('Camera On')

    def slotCameraButton(self):
        if not self.timer_camera.isActive():
            self.openCamera()
        else:
            self.closeCamera()

    def openCamera(self):
    
# попробуйте так:              ----> vvvvvvvvvvvv <------ установите свое
        self.cap = cv2.VideoCapture("mp4Video.mp4")
# или попробуйте так:
#        self.cap = cv2.VideoCapture(0, cv2.CAP_DSHOW)
        
        self.timer_camera.start(24)
        self.cameraButton.setText('Camera off')

    def closeCamera(self):
        self.timer_camera.stop()
        self.cap.release()
        self.cameraLabel.clear()
        self.cameraButton.setText('Camera On')

        
if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)
    w = CameraPageWindow()
    w.show()
    sys.exit(app.exec_())

    在此输入图像描述

    # попробуйте так:              ----> vvvvvvvvvvvv <------ установите свое
        self.cap = cv2.VideoCapture("mp4Video.mp4")

    在此输入图像描述

    # или попробуйте так:
        self.cap = cv2.VideoCapture(0, cv2.CAP_DSHOW)

    在此输入图像描述

    • 1

相关问题