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

文件夹结构

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