diff options
| -rw-r--r-- | opencv-qt-integration-2/README.md | 2 | ||||
| -rw-r--r-- | opencv-qt-integration-2/python/ImageApp.py | 81 |
2 files changed, 83 insertions, 0 deletions
diff --git a/opencv-qt-integration-2/README.md b/opencv-qt-integration-2/README.md index f96093a..3fb312a 100644 --- a/opencv-qt-integration-2/README.md +++ b/opencv-qt-integration-2/README.md @@ -5,6 +5,8 @@ This code sample shows you how to do basic image manipulation to an image and di  +The same program written in Python is available in the `python/` directory. + To compile and run the code, you need to have Qt 5 installed on your computer. The code is successfully tested on the following environment: - Mac OS X Mavericks diff --git a/opencv-qt-integration-2/python/ImageApp.py b/opencv-qt-integration-2/python/ImageApp.py new file mode 100644 index 0000000..7745711 --- /dev/null +++ b/opencv-qt-integration-2/python/ImageApp.py @@ -0,0 +1,81 @@ +#!/usr/bin/env python +# encoding: utf-8 + +import os +import sys + +try: + from PyQt5.QtCore import * + from PyQt5.QtGui import * + from PyQt5.QtWidgets import * + import cv2 +except ImportError: + print("Please install the required modules.") + sys.exit() + + +class ImageApp(QWidget): + + def __init__(self): + QWidget.__init__(self) + self.original_img = cv2.imread("../../assets/The_Chapter_House.jpg") + if self.original_img is not None: + self.original_img = cv2.cvtColor(self.original_img, cv2.COLOR_BGR2RGB) + self.setup_ui() + self.show_image(self.original_img) + + def setup_ui(self): + """Setup the UI widgets.""" + self.image_label = QLabel() + # Setup buttons + self.original_btn = QPushButton("Original") + self.original_btn.clicked.connect(self.show_original_image) + self.blur_btn = QPushButton("Gaussian Blur") + self.blur_btn.clicked.connect(self.do_gaussian_blur) + self.canny_btn = QPushButton("Canny") + self.canny_btn.clicked.connect(self.do_canny) + self.quit_btn = QPushButton("Quit") + self.quit_btn.clicked.connect(self.close) + # Setup layout for buttons + self.buttons_layout = QVBoxLayout() + self.buttons_layout.addWidget(self.original_btn) + self.buttons_layout.addWidget(self.blur_btn) + self.buttons_layout.addWidget(self.canny_btn) + self.buttons_layout.addStretch() + self.buttons_layout.addWidget(self.quit_btn) + # Setup main layout + self.main_layout = QHBoxLayout() + self.main_layout.addWidget(self.image_label) + self.main_layout.addLayout(self.buttons_layout) + self.setLayout(self.main_layout) + self.setWindowTitle("Image Processing with Qt and OpenCV") + + def show_original_image(self): + """Redraw original image.""" + self.show_image(self.original_img) + + def do_gaussian_blur(self): + """Perform Canny edge detection on original image and display the result.""" + img = cv2.GaussianBlur(self.original_img, (15, 15), 0, 0) + self.show_image(img) + + def do_canny(self): + """Perform Gaussian blurring on original image and display the result.""" + img = cv2.cvtColor(self.original_img, cv2.COLOR_RGB2GRAY) + img = cv2.Canny(img, 150, 150) + img = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB) + self.show_image(img) + + def show_image(self, img): + if img is not None: + img_ = QImage(img.data, img.shape[1], img.shape[0], QImage.Format_RGB888) + self.image_label.setPixmap(QPixmap.fromImage(img_)) + else: + self.image_label.setText("Cannot load the input image!") + + +if __name__ == "__main__": + app = QApplication(sys.argv) + image_app = ImageApp() + image_app.show() + app.exec_() |