import cv2
import numpy as np
import os
import tempfile
from PyQt5 import QtCore, QtGui, QtWidgets, uic
from PyQt5.QtGui import QPixmap
from PyQt5.QtWidgets import (QApplication, QComboBox, QDialog, QFileDialog,
QLabel, QPushButton, QStackedWidget, QVBoxLayout,
QWidget)
from PyQt5.uic import loadUi
from skimage.util import random_noise
[docs]class Ui_Dialog_3(object):
def __init__(self):
"""
Creation of a temporary image which helps in displaying the given image onto the GUI label.
"""
self.output_image = None
with tempfile.NamedTemporaryFile(suffix=".png") as f:
self.buffer_image_filename = f.name
def __del__(self):
"""
Deletion of the temporary image which helped in displaying the given image onto the GUI label.
"""
if self.output_image is not None:
os.remove(self.buffer_image_filename)
[docs] def setupUi(self, Dialog_3):
"""
setupUI defined for the screen auto-generated by the PyQt5 editor.
"""
self.counter = 0
Dialog_3.setObjectName("Dialog_3")
Dialog_3.resize(1366, 801)
self.lineEdit = QtWidgets.QLineEdit(Dialog_3)
self.lineEdit.setGeometry(QtCore.QRect(550, 660, 301, 41))
self.lineEdit.setStyleSheet("border: 12px solid rgb(0, 0, 0);\n"
"border-width: 2px;\n"
"font: 11.5pt \"MS Shell Dlg 2\";\n"
"border-radius: 15px;\n"
"padding-left:20px;\n"
"padding-right:20px;")
self.lineEdit.setText("")
self.lineEdit.setAlignment(QtCore.Qt.AlignCenter)
self.lineEdit.setObjectName("lineEdit")
self.label_5 = QtWidgets.QLabel(Dialog_3)
self.label_5.setGeometry(QtCore.QRect(400, 740, 581, 41))
self.label_5.setStyleSheet("color: rgb(255, 0, 0);")
self.label_5.setText("")
self.label_5.setAlignment(QtCore.Qt.AlignCenter)
self.label_5.setObjectName("label_5")
self.Save_As = QtWidgets.QPushButton(Dialog_3)
self.Save_As.setGeometry(QtCore.QRect(900, 660, 403, 81))
self.Save_As.setObjectName("Save_As")
self.scrollArea_2 = QtWidgets.QScrollArea(Dialog_3)
self.scrollArea_2.setGeometry(QtCore.QRect(730, 40, 571, 561))
self.scrollArea_2.setWidgetResizable(True)
self.scrollArea_2.setObjectName("scrollArea_2")
self.scrollAreaWidgetContents_2 = QtWidgets.QWidget()
self.scrollAreaWidgetContents_2.setGeometry(QtCore.QRect(0, 0, 569, 559))
self.scrollAreaWidgetContents_2.setObjectName("scrollAreaWidgetContents_2")
self.label_2 = QtWidgets.QLabel(self.scrollAreaWidgetContents_2)
self.label_2.setGeometry(QtCore.QRect(10, 20, 571, 561))
self.label_2.setText("")
self.label_2.setObjectName("label_2")
self.scrollArea_2.setWidget(self.scrollAreaWidgetContents_2)
self.Open_Image_Button = QtWidgets.QPushButton(Dialog_3)
self.Open_Image_Button.setGeometry(QtCore.QRect(90, 660, 401, 81))
self.Open_Image_Button.setObjectName("Open_Image_Button")
self.scrollArea = QtWidgets.QScrollArea(Dialog_3)
self.scrollArea.setGeometry(QtCore.QRect(80, 40, 571, 561))
self.scrollArea.setWidgetResizable(True)
self.scrollArea.setObjectName("scrollArea")
self.scrollAreaWidgetContents = QtWidgets.QWidget()
self.scrollAreaWidgetContents.setGeometry(QtCore.QRect(0, 0, 569, 559))
self.scrollAreaWidgetContents.setObjectName("scrollAreaWidgetContents")
self.label = QtWidgets.QLabel(self.scrollAreaWidgetContents)
self.label.setGeometry(QtCore.QRect(10, 20, 571, 561))
self.label.setText("")
self.label.setObjectName("label")
self.scrollArea.setWidget(self.scrollAreaWidgetContents)
self.label_3 = QtWidgets.QLabel(Dialog_3)
self.label_3.setGeometry(QtCore.QRect(660, 300, 61, 31))
self.label_3.setStyleSheet("font: 22pt \"MS Shell Dlg 2\";")
self.label_3.setAlignment(QtCore.Qt.AlignCenter)
self.label_3.setObjectName("label_3")
self.lineEdit_2 = QtWidgets.QLineEdit(Dialog_3)
self.lineEdit_2.setGeometry(QtCore.QRect(550, 710, 301, 41))
self.lineEdit_2.setStyleSheet("border: 12px solid rgb(0, 0, 0);\n"
"border-width: 2px;\n"
"font: 11.5pt \"MS Shell Dlg 2\";\n"
"border-radius: 15px;\n"
"padding-left:20px;\n"
"padding-right:20px;")
self.lineEdit_2.setText("")
self.lineEdit_2.setAlignment(QtCore.Qt.AlignCenter)
self.lineEdit_2.setObjectName("lineEdit_2")
self.retranslateUi(Dialog_3)
QtCore.QMetaObject.connectSlotsByName(Dialog_3)
[docs] def retranslateUi(self, Dialog_3):
"""
Setting the UI layouts and components - labels, textfields, pushbuttons, etc.
Also adding functionality for redirecting buttons to respective functions on being clicked.
"""
_translate = QtCore.QCoreApplication.translate
Dialog_3.setWindowTitle(_translate("Dialog_3", "Thresholding With Background"))
self.lineEdit.setPlaceholderText(_translate("Dialog_3", "Enter Lower Limit..."))
self.lineEdit_2.setPlaceholderText(_translate("Dialog_3", "Enter Upper Limit..."))
self.label_3.setText(_translate("Dialog_3", "=>"))
self.Open_Image_Button.setText(_translate("Dialog_3", "Open Image"))
self.Save_As.setText(_translate("Dialog_3", "Save As"))
self.Open_Image_Button.clicked.connect(self.File_Select)
self.Save_As.clicked.connect(self.Save_Directory)
[docs] def File_Select(self):
"""
Main Logical function which does the following:
1. Helps user in choosing input for image file (supported file formats - .jpg, .jpeg, .gif, .png)
2. Error handling if user doesn't choose supported file.
3. Displays the image onto the GUI label layout (if size exceeds, then scroll bars appear in the Label).
4. The user can Save the generated image in the directory they want on the machine.
"""
Lower_Limit = self.lineEdit.text() # Accessing the lower limit value entered by the user.
Upper_Limit = self.lineEdit_2.text() # Accessing the upper limit value entered by the user.
if not (int(Lower_Limit.isdigit() and Upper_Limit.isdigit())):
self.label_5.setText("Please enter an integer value!")
else:
self.counter += 1
self.label_5.setText("")
# fname = QFileDialog.getOpenFileName(self, "Open File", "All_Project_Files\Final_Project_Files\Cam_Media", "Images (*.png *.xpm *.jpg)")
# # Opening the Image
# self.pixmap = QPixmap(fname[0]) # This returns a tuple and hence we mention [0].
# # Adding the picture to the Label.
# self.label.setPixmap(self.pixmap)
Lower_Limit = int(self.lineEdit.text())
Upper_Limit = int(self.lineEdit_2.text())
file_name, _ = QFileDialog.getOpenFileName(None, 'Open Image File', r"<Default dir>", "Image files (*.jpg *.jpeg *.gif *.png)")
if file_name:
self.counter += 1
self.label.setPixmap(QPixmap(file_name))
img = cv2.imread(file_name, 0)
m, n = img.shape
print("The original size of the image is ", m, " x ", n)
self.output_image = []
a = Lower_Limit
b = Upper_Limit
for i in range(len(img)):
temp = []
for j in range(len(img)):
if img[i][j] > a and img[i][j] < b:
temp.append(255)
else:
temp.append(img[i][j])
self.output_image.append(temp)
self.output_image = np.array(self.output_image)
m, n= self.output_image.shape
print("The new size of the image is ", m, " x ", n)
Thresholding_With_File_Name = self.buffer_image_filename
cv2.imwrite(Thresholding_With_File_Name, self.output_image)
# self.label_2.setPixmap(QPixmap(Thresholding_With_File_Name))
lay = QtWidgets.QVBoxLayout(self.scrollAreaWidgetContents)
lay_2 = QtWidgets.QVBoxLayout(self.scrollAreaWidgetContents_2)
lay.setContentsMargins(0, 0, 0, 0)
lay_2.setContentsMargins(0, 0, 0, 0)
lay.addWidget(self.label)
lay_2.addWidget(self.label_2)
self.label.setPixmap(QPixmap(file_name))
self.label_2.setPixmap(QPixmap(Thresholding_With_File_Name))
# self.scrollArea.setWidgetResizable(True)
# self.scrollArea_2.setWidgetResizable(True)
self.label.setAlignment(QtCore.Qt.AlignHCenter | QtCore.Qt.AlignVCenter)
self.label_2.setAlignment(QtCore.Qt.AlignHCenter | QtCore.Qt.AlignVCenter)
self.Open_Image_Button.setEnabled(False)
# If you want these to display these in separate windows other than GUI.
# cv2.imshow("Negative Image", negative_img)
# cv2.imshow("Image", img)
# cv2.waitKey(0)
# # closing all open windows
# cv2.destroyAllWindows()
[docs] def Save_Directory(self):
"""
Helps in saving and writing the image in the directory the user wants to save in.
"""
if self.counter > 0:
self.label_5.setText("")
option = QFileDialog.Options()
save_as_path = QFileDialog.getSaveFileName(None, 'Open Image File', r"Thresholding With Background", "Image files (*.jpg *.jpeg *.gif *.png)")
if save_as_path[0].__len__() > 0:
cv2.imwrite(save_as_path[0], self.output_image)
else:
self.label_5.setText("Select Image first!")
if __name__ == "__main__":
import sys
app = QtWidgets.QApplication(sys.argv)
Dialog_3 = QtWidgets.QDialog()
ui = Ui_Dialog_3()
ui.setupUi(Dialog_3)
Dialog_3.show()
sys.exit(app.exec_())