update label to graphicsView

Author: JiageWang

custom/graphicsView.py

@@ -0,0 +1,60 @@
+import cv2
+
+from PyQt5.QtGui import *
+from PyQt5.QtCore import *
+from PyQt5.QtWidgets import *
+
+
+class GraphicsView(QGraphicsView):
+ def __init__(self, parent=None):
+ super(GraphicsView, self).__init__(parent=parent)
+ self._zoom = 0
+ self._empty = True
+ self._photo = QGraphicsPixmapItem()
+ self._scene = QGraphicsScene(self)
+ self._scene.addItem(self._photo)
+ self.setScene(self._scene)
+ self.setAlignment(Qt.AlignCenter) # 居中显示
+ self.setDragMode(QGraphicsView.ScrollHandDrag) # 设置拖动
+ self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
+ self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
+
+ def has_photo(self):
+ return not self._empty
+
+ def set_image(self, img):
+ self._empty = False
+ img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # bgr -> rgb
+ h, w, c = img.shape # 获取图片形状
+ image = QImage(img, w, h, 3 * w, QImage.Format_RGB888)
+ pixmap = QPixmap.fromImage(image)
+ self._photo.setPixmap(pixmap)
+
+ def fitInView(self, scale=True):
+ rect = QRectF(self._photo.pixmap().rect())
+ if not rect.isNull():
+ self.setSceneRect(rect)
+ if self.has_photo():
+ unity = self.transform().mapRect(QRectF(0, 0, 1, 1))
+ self.scale(1 / unity.width(), 1 / unity.height())
+ viewrect = self.viewport().rect()
+ scenerect = self.transform().mapRect(rect)
+ factor = min(viewrect.width() / scenerect.width(),
+ viewrect.height() / scenerect.height())
+ self.scale(factor, factor)
+ self._zoom = 0
+
+ def wheelEvent(self, event):
+ if self.has_photo():
+ if event.angleDelta().y() > 0:
+ factor = 1.25
+ self._zoom += 1
+ else:
+ factor = 0.8
+ self._zoom -= 1
+ if self._zoom > 0:
+ self.scale(factor, factor)
+ elif self._zoom == 0:
+ self.fitInView()
+ else:
+ self._zoom = 0

custom/listWidgetItems.py

@@ -35,7 +35,7 @@ def __call__(self, img):
 class FilterItem(MyItem):
 
  def __init__(self, parent=None):
- super().__init__('平滑操作', parent=parent)
+ super().__init__('平滑处理', parent=parent)
  self._ksize = 3
  self._kind = MEAN_FILTER
  self._sigmax = 0
@@ -52,7 +52,7 @@ def __call__(self, img):
 
 class MorphItem(MyItem):
  def __init__(self, parent=None):
- super().__init__('形态操作', parent=parent)
+ super().__init__(' 形态学 ', parent=parent)
  self._ksize = 3
  self._kind = ERODE_MORPH
  self._kshape = RECT_MORPH_SHAPE
@@ -66,7 +66,7 @@ def __call__(self, img):
 class GradItem(MyItem):
 
  def __init__(self, parent=None):
- super().__init__('梯度操作', parent=parent)
+ super().__init__('图像梯度', parent=parent)
  self._kind = SOBEL_GRAD
  self._ksize = 3
  self._dx = 1
@@ -75,11 +75,11 @@ def __init__(self, parent=None):
  def __call__(self, img):
  if self._dx == 0 and self._dy == 0 and self._kind != LAPLACIAN_GRAD:
  self.setBackground(QColor(255, 0, 0))
- self.setText('梯度操作 （无效: dx与dy不同时为0）')
+ self.setText('图像梯度 （无效: dx与dy不同时为0）')
  return img
  else:
  self.setBackground(QColor(200, 200, 200))
- self.setText('梯度操作')
+ self.setText('图像梯度')
  if self._kind == SOBEL_GRAD:
  return cv2.Sobel(img, -1, self._dx, self._dy, self._ksize)
  elif self._kind == SCHARR_GRAD:
@@ -90,7 +90,7 @@ def __call__(self, img):
 
 class ThresholdItem(MyItem):
  def __init__(self, parent=None):
- super().__init__('阈值操作', parent=parent)
+ super().__init__('阈值处理', parent=parent)
  self._thresh = 127
  self._maxval = 255
  self._type = BINARY_THRESH

main.py

@@ -12,15 +12,10 @@ class MyApp(QMainWindow, Ui_MainWindow):
  def __init__(self):
  super(MyApp, self).__init__()
  self.setupUi(self)
+ self.setWindowTitle('Opencv图像处理')
+ self.setWindowIcon(QIcon('icons/main.png'))
  self.src_img = None
- self.label.setAlignment(Qt.AlignCenter)
- self.label.setStyleSheet(
- """
- QLabel{
- background: #4B4B4B
- }
- """
- )
+ self.statusbar.close()
  self.stackedWidget.close()
 
  def update_label(self):
@@ -30,15 +25,16 @@ def update_label(self):
  for i in range(self.useListWidget.count()):
  img = self.useListWidget.item(i)(img)
 
- img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # bgr -> rgb
- h, w, c = img.shape # 获取图片形状
- image = QImage(img, w, h, 3 * w, QImage.Format_RGB888)
- pixmap = QPixmap.fromImage(image)
-
- self.item = QGraphicsPixmapItem(pixmap) # 创建像素图元
- self.scene = QGraphicsScene() # 创建场景
- self.scene.addItem(self.item)
- self.label.setScene(self.scene)
+ # img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # bgr -> rgb
+ # h, w, c = img.shape # 获取图片形状
+ # image = QImage(img, w, h, 3 * w, QImage.Format_RGB888)
+ # pixmap = QPixmap.fromImage(image)
+ #
+ # self.item = QGraphicsPixmapItem(pixmap) # 创建像素图元
+ # self.scene = QGraphicsScene() # 创建场景
+ # self.scene.addItem(self.item)
+ # self.graphicsView.setScene(self.scene)
+ self.graphicsView.set_image(img)
 
 
 if __name__ == "__main__":

windows/main_window.py

@@ -10,6 +10,7 @@
 from custom.tableWidget import *
 from custom.listWidgets import *
 from custom.treeView import *
+from custom.graphicsView import *
 
 
 class Ui_MainWindow(object):
@@ -34,15 +35,15 @@ def setupUi(self, MainWindow):
  self.treeView.setMinimumSize(QtCore.QSize(200, 0))
  self.treeView.setObjectName("treeView")
  self.horizontalLayout.addWidget(self.treeView)
- self.label = QGraphicsView(self.centralwidget)
+ self.graphicsView = GraphicsView(self.centralwidget)
  sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Expanding)
- sizePolicy.setHorizontalStretch(0)
+ sizePolicy.setHorizontalStretch(1)
  sizePolicy.setVerticalStretch(0)
- sizePolicy.setHeightForWidth(self.label.sizePolicy().hasHeightForWidth())
- self.label.setSizePolicy(sizePolicy)
- self.label.setMinimumSize(QtCore.QSize(1280, 960))
- self.label.setObjectName("label")
- self.horizontalLayout.addWidget(self.label)
+ sizePolicy.setHeightForWidth(self.graphicsView.sizePolicy().hasHeightForWidth())
+ self.graphicsView.setSizePolicy(sizePolicy)
+ self.graphicsView.setMinimumSize(QtCore.QSize(640, 480))
+ self.graphicsView.setObjectName("label")
+ self.horizontalLayout.addWidget(self.graphicsView)
  self.verticalLayout = QtWidgets.QVBoxLayout()
  self.verticalLayout.setObjectName("verticalLayout")
  self.useListWidget = UsedListWidget(self.centralwidget)