這是我在 Stack Overflow 上的第一篇文章,所以我希望我做的一切都是正確的。
我一直在嘗試在使用 PyQt5 制作的 GUI 上使用 OpenCV 獲取視頻源。我最終讓它作業了,但它總是 640 x 480(我設定的大小)。但是,我希望它自動縮放/調整大小(具有相同的縱橫比)以適應其育兒小部件。例如,當我展開視窗時,視頻源也應該縮放。有誰知道這樣做的方法?
這是我的代碼示例以供參考:
相機小部件
class Camera(QWidget):
def __init__(self, port):
super().__init__()
self.display_width = 640
self.display_height = 480
self.camera = QLabel(self)
self.camera.resize(self.display_width, self.display_height)
self.label = QLabel(f'Port {port}')
self.label.setStyleSheet("""
QLabel {
color: white;
font: bold 30px
}
""")
# Layout
self.layout = QVBoxLayout()
self.layout.addWidget(self.label)
self.layout.addWidget(self.camera)
self.setLayout(self.layout)
self.thread = VideoThread(port)
self.thread.change_pixmap_signal.connect(self.update_image)
self.thread.start()
def close_event(self, event):
self.thread.stop()
event.accept()
@pyqtSlot(ndarray)
def update_image(self, cv_img):
qt_img = self.convert_cv_qt(cv_img)
self.camera.setPixmap(qt_img)
def convert_cv_qt(self, cv_img):
rgb_image = cv2.cvtColor(cv_img, cv2.COLOR_BGR2RGB)
h, w, ch = rgb_image.shape
bytes_per_line = ch * w
convert_to_Qt_format = QImage(rgb_image.data, w, h, bytes_per_line, QImage.Format_RGB888)
p = convert_to_Qt_format.scaled(self.display_width, self.display_height, Qt.KeepAspectRatio)
return QPixmap.fromImage(p)
視頻源執行緒
class VideoThread(QThread):
change_pixmap_signal = pyqtSignal(ndarray)
def __init__(self, port):
super().__init__()
self.running = True
self.port = port
def run(self):
self.capture = cv2.VideoCapture(self.port)
while self.running:
self.ret, self.image = self.capture.read()
if self.ret:
self.change_pixmap_signal.emit(self.image)
self.capture.release()
def stop(self):
self.running = False
self.wait()
uj5u.com熱心網友回復:
您應該獲得 resize 事件self.camera并將其寬度設定為self.display_width,高度設定為self.display_height
這個陳述句p = convert_to_Qt_format.scaled(self.display_width, self.display_height, Qt.KeepAspectRatio) 將完成剩下的作業
更新
設定最小尺寸以避免遞回回圈,這將繼續擴大視窗
為避免搖晃,您可以使用 afixed size container或 removeQt.KeepAspectRatio這有助于搖晃窗戶
- 這是更新的代碼
from PyQt5 import QtCore, QtGui, QtWidgets
import cv2
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from PyQt5.QtCore import *
from numpy import ndarray
class Camera(QWidget):
def __init__(self, port):
super().__init__()
self.display_width = 640
self.display_height = 480
self.camera = QLabel()
# self.camera.setGeometry(0,0,640,480)
self.camera.resize(self.display_width, self.display_height)
# Set Minimum size of camera stream to avoid going in recursive loop
self.camera.setMinimumWidth(640)
self.camera.setMinimumHeight(480)
self.label = QLabel(f'Port {port}')
self.label.setStyleSheet("""
color: black;
font: 30px;
""")
# Layout
self.layout = QVBoxLayout()
self.layout.addWidget(self.label)
self.layout.addWidget(self.camera)
self.setLayout(self.layout)
# Get the resize Event Callback
self.resizeEvent = self.label_resize
self.camera.resizeEvent = self.camera_resize
self.thread = VideoThread(port)
self.thread.change_pixmap_signal.connect(self.update_image)
self.thread.start()
# Resize Event Callback
def label_resize(self, resizeEvent:QResizeEvent):
self.camera.resize(resizeEvent.size())
# Resize Event Callback
def camera_resize(self, resizeEvent:QResizeEvent):
self.display_width, self.display_height = self.camera.width(), self.camera.height()
def close_event(self, event):
self.thread.stop()
event.accept()
@pyqtSlot(ndarray)
def update_image(self, cv_img):
qt_img = self.convert_cv_qt(cv_img)
self.camera.setPixmap(qt_img)
def convert_cv_qt(self, cv_img):
rgb_image = cv2.cvtColor(cv_img, cv2.COLOR_BGR2RGB)
h, w, ch = rgb_image.shape
bytes_per_line = ch * w
convert_to_Qt_format = QImage(rgb_image.data, w, h, bytes_per_line, QImage.Format_RGB888)
p = convert_to_Qt_format.scaled(self.display_width, self.display_height, Qt.KeepAspectRatio)
return QPixmap.fromImage(p)
class VideoThread(QThread):
change_pixmap_signal = pyqtSignal(ndarray)
def __init__(self, port):
super().__init__()
self.running = True
self.port = port
def run(self):
self.capture = cv2.VideoCapture(self.port)
while self.running:
self.ret, self.image = self.capture.read()
if self.ret:
self.change_pixmap_signal.emit(self.image)
self.capture.release()
def stop(self):
self.running = False
self.wait()
轉載請註明出處,本文鏈接:https://www.uj5u.com/yidong/432792.html
上一篇:OpenCV:將向量堆疊到墊子上
