mobilenet分类，pyqt5界面显示，记录

最新推荐文章于 2024-06-24 11:46:56 发布

常山ii

最新推荐文章于 2024-06-24 11:46:56 发布

阅读量601

点赞数

文章标签：分类 python qt

本文链接：https://blog.youkuaiyun.com/qq_41619994/article/details/124775177

版权

这篇博客记录了使用PyQT5创建界面显示MobileNet分类结果的过程。作者在myMainWin.py中调用predict_new函数时遇到了参数传递错误，通过修正将图片路径读取为Image对象解决了问题。文中提到，初始化时不需设置label，而应在相关动作函数中初始化。作者以初学者的身份分享学习心得，并引用了相关教程资源。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文件夹结构

mobilenet{
		mainwin.ui
		mainwin.py
		img.jpg		预测的图片
		class.json
		mobilenet_v2.pth		预训练权重
		MobileNetV2.pth		自己的数据集训练好的权重
		mobilenet_v2.py
		myMainWin.py		编写调用窗口程序
		predict.py		预测
		train.py		训练
}

其中myMainWin.py代码如下

import sys
from PyQt5.QtWidgets import QMainWindow, QApplication, QFileDialog, QLabel
from PyQt5.QtGui import QIcon, QImage, QPixmap
from mainwin import Ui_MainWindow
import cv2
import numpy
from predict_new_v2 import predict_new
from PIL import Image


class myMainWin(QMainWindow, Ui_MainWindow):
    def __init__(self):
        super(myMainWin, self).__init__()
        self.setupUi(self)  

        # 设置主窗口的标题
        self.setWindowTitle('基于xxx分类系统')

        # 连接动作对应的函数

        self.pushButton_2.clicked.connect(self.openimg)  # 构造函数（label可以不用写）
        self.pushButton_3.clicked.connect(self.detect)

    def openimg(self):
        global fname
        # 选择且获取图片文件的地址
        fileName, filetype = QFileDialog.getOpenFileName(
            self

最低0.47元/天解锁文章