Question

main.py：

from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtGui import QPixmap, QImage
import cv2
import os
class Ui_MainWindow(object):
    def setupUi(self, MainWindow):

        self.loadimg.clicked.connect(self.load_img)

    def load_img(self):
        load_img_1, filter = QtWidgets.QFileDialog.getOpenFileName(None, "Select Image",
                                                                    options=QtWidgets.QFileDialog.DontUseNativeDialog)
        img = cv2.imread(load_img_1)

        if img is None:
            self.imglabel.setText("Cannot load the input image.")
        else:
            img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
            img_ = QImage(img.data, img.shape[1], img.shape[0], QImage.Format_RGB888)
            self.imglabel.setPixmap(QPixmap.fromImage(img_))
            pixmap = QPixmap.fromImage(img_)
            pixmap = pixmap.scaled(self.imglabel.width(), self.imglabel.height())
        if load_img_1:
            pixmap = QtGui.QPixmap(load_img_1)
            pixmap = pixmap.scaled(self.imglabel.width(), self.imglabel.height(),QtCore.Qt.KeepAspectRatio)
            self.imglabel.setPixmap(pixmap)
            self.imglabel.setAlignment(QtCore.Qt.AlignCenter)
            self.imglabel.setScaledContents(True)

if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    MainWindow = QtWidgets.QMainWindow()
    ui = Ui_MainWindow()
    ui.setupUi(MainWindow)
    MainWindow.show()
    sys.exit(app.exec_())

classify.py：

image = cv2.imread("/")

我是python和机器学习的初学者。我的学习有问题。我在main.py中制作了一个GUI，并且通过单击GUI def load_img(self):上的按钮加载图像时，我希望图像的路径链接到image = cv2.imread("/")中的classify.py，但是我不知道不知道该怎么做。此外，我想如何在GUI的文本标签上显示分类结果，我也不知道如何。因此，我需要您的帮助，并向我展示如何做到这一点。非常感谢。

Answer 1

在classify.py中，将代码放入带有参数的函数中。 path and then you can导入it and run function with your image filename as parametr. If function will return image then main`可以获取并显示。

classify.py

def run(path)

    image = cv2.imread(path)

    # ... machine learning code ...

    return image

main.py-请参见import和image = clasify.run(filename)

from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtGui import QPixmap, QImage
import cv2
import os

import classify # filename without .py

class Ui_MainWindow(object):
    def setupUi(self, MainWindow):

        self.loadimg.clicked.connect(self.load_img)

    def load_img(self):
        load_img_1, filter = QtWidgets.QFileDialog.getOpenFileName(None, "Select Image",
                                                                    options=QtWidgets.QFileDialog.DontUseNativeDialog)

        #img = cv2.imread(load_img_1)
        img = classify.run(load_img_1) # use it


        if img is None:
            self.imglabel.setText("Cannot load the input image.")
        else:
            img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
            img_ = QImage(img.data, img.shape[1], img.shape[0], QImage.Format_RGB888)
            self.imglabel.setPixmap(QPixmap.fromImage(img_))
            pixmap = QPixmap.fromImage(img_)
            pixmap = pixmap.scaled(self.imglabel.width(), self.imglabel.height())
        if load_img_1:
            pixmap = QtGui.QPixmap(load_img_1)
            pixmap = pixmap.scaled(self.imglabel.width(), self.imglabel.height(),QtCore.Qt.KeepAspectRatio)
            self.imglabel.setPixmap(pixmap)
            self.imglabel.setAlignment(QtCore.Qt.AlignCenter)
            self.imglabel.setScaledContents(True)

if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    MainWindow = QtWidgets.QMainWindow()
    ui = Ui_MainWindow()
    ui.setupUi(MainWindow)
    MainWindow.show()
    sys.exit(app.exec_())

如何通过单击GUI上的按钮将图像获取到同一目录文件夹中的另一个.py文件中

1 个答案: