# -*- coding: utf-8 -*-
import os
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import *
import tensorflow as tf
from PIL import Image
import numpy as np
import sys
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(800, 600)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
self.label = QtWidgets.QLabel(self.centralwidget)
self.label.setGeometry(QtCore.QRect(220, 20, 291, 61))
self.label.setScaledContents(False)
self.label.setObjectName("label")
self.pushButton = QtWidgets.QPushButton(self.centralwidget)
self.pushButton.setGeometry(QtCore.QRect(160, 430, 93, 28))
self.pushButton.setObjectName("pushButton")
self.pushButton_2 = QtWidgets.QPushButton(self.centralwidget)
self.pushButton_2.setGeometry(QtCore.QRect(440, 430, 93, 28))
self.pushButton_2.setObjectName("pushButton_2")
self.label_2 = QtWidgets.QLabel(self.centralwidget)
self.label_2.setGeometry(QtCore.QRect(150, 90, 381, 321))
self.label_2.setText("")
self.label_2.setObjectName("label_2")
self.label_3 = QtWidgets.QLabel(self.centralwidget)
self.label_3.setGeometry(QtCore.QRect(550, 130, 141, 51))
self.label_3.setText("")
self.label_3.setObjectName("label_3")
self.label_4 = QtWidgets.QLabel(self.centralwidget)
self.label_4.setGeometry(QtCore.QRect(550, 90, 141, 31))
self.label_4.setObjectName("label_4")
self.textBrowser = QtWidgets.QTextBrowser(self.centralwidget)
self.textBrowser.setGeometry(QtCore.QRect(150, 90, 381, 321))
self.textBrowser.setObjectName("textBrowser")
self.textBrowser_2 = QtWidgets.QTextBrowser(self.centralwidget)
self.textBrowser_2.setGeometry(QtCore.QRect(550, 130, 141, 51))
self.textBrowser_2.setObjectName("textBrowser_2")
self.textBrowser_3 = QtWidgets.QTextBrowser(self.centralwidget)
self.textBrowser_3.setGeometry(QtCore.QRect(220, 20, 291, 61))
self.textBrowser_3.setObjectName("textBrowser_3")
self.textBrowser_4 = QtWidgets.QTextBrowser(self.centralwidget)
self.textBrowser_4.setGeometry(QtCore.QRect(550, 90, 141, 31))
self.textBrowser_4.setObjectName("textBrowser_4")
self.textBrowser_2.raise_()
self.label.raise_()
self.textBrowser.raise_()
self.textBrowser_3.raise_()
self.pushButton.raise_()
self.pushButton_2.raise_()
self.label_2.raise_()
self.label_4.raise_()
self.textBrowser_4.raise_()
self.label_3.raise_()
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtWidgets.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 26))
self.menubar.setObjectName("menubar")
MainWindow.setMenuBar(self.menubar)
self.statusbar = QtWidgets.QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
MainWindow.setStatusBar(self.statusbar)
self.toolBar = QtWidgets.QToolBar(MainWindow)
self.toolBar.setObjectName("toolBar")
MainWindow.addToolBar(QtCore.Qt.TopToolBarArea, self.toolBar)
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
# 模型相关变量初始化
self.model = tf.keras.models.load_model(r'C:\Users\zjl15\PycharmProjects\pythonProject1\roch_classification_cnn.h5')
self.path = ''
self.rock_types = ['砾岩','安山岩','花岗岩','石灰岩','石英岩']
# 将“导入图片”按钮与openImage函数绑定
self.pushButton.clicked.connect(self.openImage)
# 将“岩石分类”按钮与classify函数绑定
self.pushButton_2.clicked.connect(self.classify)
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
self.label.setText(_translate("MainWindow", "岩石图像分类"))
self.pushButton.setText(_translate("MainWindow", "导入图像"))
self.pushButton_2.setText(_translate("MainWindow", "岩石分类"))
self.label_4.setText(_translate("MainWindow", "分类结果"))
self.textBrowser_3.setHtml(_translate("MainWindow",
"<!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0//EN\" \"http://www.w3.org/TR/REC-html40/strict.dtd\">\n"
"<html><head><meta name=\"qrichtext\" content=\"1\" /><style type=\"text/css\">\n"
"p, li { white-space: pre-wrap; }\n"
"</style></head><body style=\" font-family:\'SimSun\'; font-size:9pt; font-weight:400; font-style:normal;\">\n"
"<p align=\"center\" style=\" margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;\"><span style=\" font-size:24pt;\">岩石图像识别</span></p></body></html>"))
self.textBrowser_4.setHtml(_translate("MainWindow",
"<!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0//EN\" \"http://www.w3.org/TR/REC-html40/strict.dtd\">\n"
"<html><head><meta name=\"qrichtext\" content=\"1\" /><style type=\"text/css\">\n"
"p, li { white-space: pre-wrap; }\n"
"</style></head><body style=\" font-family:\'SimSun\'; font-size:9pt; font-weight:400; font-style:normal;\">\n"
"<p align=\"center\" style=\" margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;\"><span style=\" font-size:11pt;\">分类结果</span></p></body></html>"))
self.toolBar.setWindowTitle(_translate("MainWindow", "toolBar"))
# 导入图片函数
def resource_path(relative):
if hasattr(sys, "_MEIPASS"):
absolute_path = os.path.join(sys._MEIPASS, relative)
else:
absolute_path = os.path.join(relative)
return absolute_path
# 在原来引用该文件的地方加上这个函数 (resource_path("文件名"))
def openImage(self):
imgPath, imgType = QFileDialog.getOpenFileName(None, "导入图片", "", "*.jpg;;*.png;;All Files(*)")
jpg = QtGui.QPixmap(imgPath).scaled(self.label_2.width(), self.label_2.height())
self.label_2.setPixmap(jpg)
self.path=imgPath
self.label_3.setText('')
def classify(self):
img = Image.open(self.path) # 读取图像
img_rgb = img.split()
x = np.zeros((1, 224, 224, 3))
x[0,:, :, 0] = np.array(img_rgb[0]) / 255
x[0,:, :, 1] = np.array(img_rgb[1]) / 255
x[0,:, :, 2] = np.array(img_rgb[2]) / 255
y = self.model.predict(x)
result = self.rock_types[np.argmax(y)]
self.label_3.setText(result)
if __name__=='__main__':
QtCore.QCoreApplication.setAttribute(QtCore.Qt.AA_EnableHighDpiScaling)
app=QtWidgets.QApplication(sys.argv)
MainWindow=QtWidgets.QMainWindow()
ui_test=Ui_MainWindow()
ui_test.setupUi(MainWindow)
MainWindow.show()
sys.exit(app.exec_())
pythonProject1.rar
打开即用 （1776个子文件） 
acc_11_final_mini.csv 533.05MB acc_10_final_mini.csv 520.94MB acc_09_final_mini.csv 489.91MB acc_08_final_mini.csv 451.68MB air_data.csv 13.56MB .gitignore 184B roch_classification_cnn.h5 277.59MB
report.html 3.48MB pythonProject1.iml 291B 2.ipynb 2.48MB 2_2.ipynb 2.25MB 4.ipynb 1.84MB 3_2.ipynb 335KB 1.ipynb 304KB 3.ipynb 260KB
0_32_noisy.jpg 53KB 1_56_noisy.jpg 52KB 2_44_noisy.jpg 52KB 0_32.jpg 49KB 0_48_noisy.jpg 49KB 2_34_noisy.jpg 48KB 0_32_contrast.jpg 48KB 2_32_noisy.jpg 48KB 0_32.jpg 47KB 0_32_flip.jpg 47KB 2_26_noisy.jpg 47KB 2_35_noisy.jpg 47KB 2_54_noisy.jpg 47KB 3_42_noisy.jpg 47KB 2_44_contrast.jpg 46KB 2_51_noisy.jpg 46KB 2_57_noisy.jpg 46KB 2_53_noisy.jpg 46KB 2_43_noisy.jpg 45KB 2_33_noisy.jpg 45KB 0_6_noisy.jpg 45KB 0_1_noisy.jpg 45KB 2_42_noisy.jpg 44KB 0_60_noisy.jpg 44KB 2_34_contrast.jpg 44KB 2_44.jpg 44KB 2_44.jpg 44KB 2_41_noisy.jpg 44KB 2_44_flip.jpg 44KB 2_35_contrast.jpg 44KB 1_56_contrast.jpg 44KB 3_32_noisy.jpg 44KB 0_41_noisy.jpg 43KB 0_44_noisy.jpg 43KB 0_33_noisy.jpg 43KB 1_56.jpg 43KB 1_56_flip.jpg 43KB 2_26_contrast.jpg 43KB 1_56.jpg 43KB 3_36_noisy.jpg 43KB 0_22_noisy.jpg 43KB 2_34.jpg 43KB 2_34.jpg 43KB 2_34_flip.jpg 43KB 0_25_noisy.jpg 43KB 3_39_noisy.jpg 43KB 2_30_noisy.jpg 43KB 2_56_noisy.jpg 43KB 1_39_noisy.jpg 43KB 2_31_noisy.jpg 43KB 2_36_noisy.jpg 43KB 2_37_noisy.jpg 43KB 3_23_noisy.jpg 43KB 2_22_noisy.jpg 43KB 2_26.jpg 42KB 0_26_noisy.jpg 42KB 1_12_noisy.jpg 42KB 0_55_noisy.jpg 42KB 1_36_noisy.jpg 42KB 2_26.jpg 42KB 3_35_noisy.jpg 42KB 2_26_flip.jpg 42KB 0_47_noisy.jpg 42KB 2_40_noisy.jpg 42KB 2_55_noisy.jpg 42KB 2_33.jpg 42KB 3_48_noisy.jpg 42KB 3_40_noisy.jpg 42KB 0_29_noisy.jpg 42KB 2_29_noisy.jpg 42KB 2_35.jpg 42KB 2_43_contrast.jpg 42KB 4_60_noisy.jpg 41KB 1_57_noisy.jpg 41KB 2_33.jpg 41KB 2_33_flip.jpg 41KB 2_35_flip.jpg 41KB 2_35.jpg 41KB 1_10_noisy.jpg 41KB 4_8_noisy.jpg 41KB 2_28_noisy.jpg 41KB 0_50_noisy.jpg 41KB 2_59_noisy.jpg 41KB 0_27_noisy.jpg 41KB 0_32_rotated.jpg 41KB共 1776 条
- 1
- 2
- 3
- 4
- 5
- 6
- 18
资源评论
张謹礧
- 粉丝: 3w+
最新资源
- 基于COMSOL多物理场的隧洞开挖流固耦合模型：应力影响下的土体变形与渗透特性分析 · 多物理场建模 必备版
- 光储联合离网微电网：光伏MPPT扰动观察法稳定直流母线电压仿真分析
- 企业级STM32 Boot Loader：优化验证后的实用代码包与QT上位机源码详解 - Flash编程
- 01Studio CanMV K230 开发板，单路摄像头显示，默认外接HDMI显示器，也可以使用3.5寸触摸屏显示
- 两轮四轮差速机器人STM32底层源码与ROS端工程源码：实现高精度定位与导航的融合算法 · EKF
- 高效工业相机与机器视觉软件：AI驱动的轴承保持架缺陷快速检测系统，实时采集与通讯，漏检率低于1%
- 5G数字电源方案：基于无桥PFC三相交错零电压模式的6.5kW高效电源设计及其实现
- 针对目标检测做的数据增强
- 光子学与微电子学中Lumerical FDTD Mode建模及特殊图案GDS版图设计的综合研究 · 微电子学
- COMSOL模拟沸腾水中气泡运动的两相流流体传热与蒸汽冷凝：模型及参数设置
- 基于MATLABSimulink的永磁同步电机无差拍电流预测控制仿真研究与实现
- 01Studio CanMV K230 开发板，双路摄像头显示程序 ，CSI1与CSI2接sener摄像头，外接HDMI显示器
- 电力系统领域：基于Matlab的配电网故障重构二阶锥优化方法及其应用
- 云广直流输电的PSCAD模型 - 高压直流输电 指南
- 轻量级目标检测 deeposrt目标追踪
- 单相七电平级联逆变器开环仿真的MATLAB Simulink实现及其应用
资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！
点击此处反馈
