Пример #1
0
class Ui_MainWindow(object):
    def setupUi(self, MainWindow):

        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(570, 351)

        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")

        self.selectImageBtn = QtWidgets.QPushButton(self.centralwidget)
        self.selectImageBtn.setGeometry(QtCore.QRect(30, 300, 93, 28))
        self.selectImageBtn.setObjectName("selectImageBtn")

        self.imageLabel = QtWidgets.QLabel(self.centralwidget)
        self.imageLabel.setGeometry(QtCore.QRect(20, 10, 531, 261))
        self.imageLabel.setFrameShape(QtWidgets.QFrame.Box)
        self.imageLabel.setText("")
        self.imageLabel.setObjectName("imageLabel")

        self.submitBtn = QtWidgets.QPushButton(self.centralwidget)
        self.submitBtn.setGeometry(QtCore.QRect(460, 300, 93, 28))
        self.submitBtn.setObjectName("submitBtn")

        self.thresholdDec = QtWidgets.QPushButton(self.centralwidget)
        self.thresholdDec.setGeometry(QtCore.QRect(150, 300, 31, 28))
        self.thresholdDec.setObjectName("thresholdDec")

        self.thresholdInc = QtWidgets.QPushButton(self.centralwidget)
        self.thresholdInc.setGeometry(QtCore.QRect(250, 300, 31, 28))
        self.thresholdInc.setObjectName("thresholdInc")

        self.thresholdVal = QtWidgets.QLineEdit(self.centralwidget)
        self.thresholdVal.setGeometry(QtCore.QRect(190, 300, 51, 31))
        self.thresholdVal.setObjectName("thresholdVal")
        self.thresholdVal.isReadOnly()
        self.thresholdVal.setAlignment(QtCore.Qt.AlignCenter)
        self.thresholdVal.setReadOnly(True)
        self.thresholdVal.setText("5")

        self.autoSegmentBtn = QtWidgets.QPushButton(self.centralwidget)
        self.autoSegmentBtn.setGeometry(QtCore.QRect(320, 300, 111, 28))
        self.autoSegmentBtn.setObjectName("autoSegmentBtn")

        MainWindow.setCentralWidget(self.centralwidget)

        self.filePath = ""
        #setting buttons to be disabled at the beginning
        self.submitBtn.setEnabled(False)
        self.thresholdDec.setEnabled(False)
        self.thresholdInc.setEnabled(False)
        self.thresholdVal.setEnabled(False)
        self.autoSegmentBtn.setEnabled(False)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

        self.selectImageBtn.clicked.connect(self.setImage)
        self.thresholdInc.clicked.connect(self.increaseThreshold)
        self.thresholdDec.clicked.connect(self.decreaseThreshold)
        self.autoSegmentBtn.clicked.connect(self.autoSegment)
        #self.submitBtn.clicked.connect(self.uploadImage)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        self.selectImageBtn.setText(_translate("MainWindow", "Select Image"))
        self.submitBtn.setText(_translate("MainWindow", "Submit"))
        self.thresholdDec.setText(_translate("MainWindow", "▼"))
        self.thresholdInc.setText(_translate("MainWindow", "▲"))
        self.autoSegmentBtn.setText(_translate("MainWindow", "Auto-Segment"))

    def setImage(self):
        filePath, _ = QtWidgets.QFileDialog.getOpenFileName(
            None, "Select Image", "", "Image Files (*.png *.jpg *.jpeg *.bmp)")
        if filePath:
            pixmap = QtGui.QPixmap(filePath)
            pixmap = pixmap.scaled(self.imageLabel.width(),
                                   self.imageLabel.height(),
                                   QtCore.Qt.KeepAspectRatio)
            self.imageLabel.setPixmap(pixmap)
            self.imageLabel.setAlignment(QtCore.Qt.AlignCenter)
            self.thresholdInc.setEnabled(True)
            self.thresholdDec.setEnabled(True)
            self.thresholdVal.setEnabled(True)
            self.autoSegmentBtn.setEnabled(True)
            self.selectImageBtn.setEnabled(False)
        self.filePath = filePath
        self.segmenter = Segmenter(filePath)

    def uploadImage(self):
        pass

    def increaseThreshold(self):
        val = (int)(self.thresholdVal.text())
        self.thresholdVal.setText(str(val + 1))
        val = val + 1
        self.segmenter.threshold_and_morph(val)

    def decreaseThreshold(self):
        val = (int)(self.thresholdVal.text())
        if val == 1:
            return
        self.thresholdVal.setText(str(val - 1))
        val = val - 1
        self.segmenter.threshold_and_morph(val)

    def autoSegment(self):
        self.segmenter.auto_segment((int)(self.thresholdVal.text()))
        self.showdialog()

    def showdialog(self):
        msg = QtWidgets.QMessageBox()
        msg.about(self.centralwidget, "Done", "Success!")
Пример #2
0
def test():
    s = Segmenter('test.png')

    s.threshold_and_morph(11)

    s.auto_segment()