summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorNash <nash@bsd-noobz.com>2014-08-31 17:28:19 +0700
committerNash <nash@bsd-noobz.com>2014-08-31 17:28:19 +0700
commit0dd600919f01a952a1396642eb864c7cd23d6150 (patch)
tree24cf9500efc07af9286dc6ecb943320ac1fea054
parent723be3d9d0bfd5841610d7395571ba30633634b1 (diff)
added python version
-rw-r--r--opencv-qt-integration-2/README.md2
-rw-r--r--opencv-qt-integration-2/python/ImageApp.py81
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
![screenshot](http://i.imgur.com/OPCiEw8.png)
+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_()