【AI达人特训营】PPOCR：电表检测模型的Windows端部署实现

# 安装PaddlePaddle，PaddleOCR，PyQt5 !pip install PaddlePaddle !pip install paddleocr !pip install PyQt5

import sys import os import csv import time import window from PyQt5 import QtWidgets from PyQt5.QtWidgets import QApplication, QMainWindow from PyQt5.QtCore import pyqtSignal, QObject from paddleocr import PaddleOCR class MySignals(QObject):     # 定义信号     signal_err_ocr = pyqtSignal(str)     signal_xz1 = pyqtSignal(str)     signal_xz2 = pyqtSignal(str) class Signals(QMainWindow, window.Ui_Form):     def __init__(self):         super().__init__()         self.setupUi(self)         self.init_ui()         # 实例化         self.ms_signals_err_ocr = MySignals()         self.ms_signals_xz1 = MySignals()         self.ms_signals_xz2 = MySignals()         # 连接信号         self.ms_signals_err_ocr.signal_err_ocr.connect(self.update_err_ocr)         self.ms_signals_xz1.signal_xz1.connect(self.update_xz1)         self.ms_signals_xz2.signal_xz2.connect(self.update_xz2)     # 更新第三个文本框     def update_err_ocr(self, err):         self.lineEdit_3.setText(err)     # 更新第一个文本框     def update_xz1(self, txt):         self.lineEdit.setText(txt.replace('/', '\'))     # 更新第二个文本框     def update_xz2(self, txt):         self.lineEdit_2.setText(txt.replace('/', '\'))     def onclick1(self):         # 输入地址         lineEdit = self.lineEdit.text()         # 输出地址         lineEdit_2 = self.lineEdit_2.text()         # 将图片文件夹内所有图片的名字、地址存储到两个列表中，方便后续使用         # 图片文件夹位置         readpath = lineEdit         # 图片名 列表         list_picturename = os.listdir(readpath)         # 图片地址 列表         list_picturedir = []         self.ms_signals_err_ocr.signal_err_ocr.emit("正在进行电表检测...")         for picturename in list_picturename:             list_picturedir.append(readpath + '/' + picturename)         # 将准备好的电表检测模型 替换掉 PaddleOCR 中的原有模型         ocr = PaddleOCR(# use_pdserving=False,                         det_model_dir='C:/Users/guay_/exercise_/exercise_job_wanted/det_finetune/det_db',  # 地址务必修改正确                         # rec_model_dir='C:/Users/guay_/exercise_/exercise_job_wanted/rec_finetune/rec_crnn',                         det=True,                         cls=True,                         # use_gpu=gpu,                         # show_log=False,                         # box_thresh=0.1,                         # det_db_box_thresh=0.2,                         use_angle_cls=True,                         lang='en')         # 电表读数         list_dbds = []         # 电表编号         list_dbbh = []         # 遍历每一张图片，运用模型进行推理         for picturedir in list_picturedir:             temp = ocr.ocr(picturedir, cls=True)             # 若未检测出，则赋值为'null'             if not temp:                 list_dbds.append('null')                 list_dbbh.append('null')             # 若只检测出电表读数             elif len(temp)==1:                 list_dbds.append(temp[0][1][0])                 list_dbbh.append('null')             else:                 list_dbds.append(temp[0][1][0])                 list_dbbh.append(temp[1][1][0])         # 获取当前时间戳         time_stamp = time.strftime('%Y-%m-%d-%H-%M-%S', time.localtime(time.time()))         # 创建文件夹如果不存在的话         savepath = lineEdit_2         if not os.path.exists(savepath):             os.makedirs(savepath)         final_csv_path = savepath + "/result_" + time_stamp + ".csv"         # 将两个列表中的数据写入一个.csv文件中         with open(final_csv_path, "w", encoding="utf-8", newline='') as f:             writer = csv.writer(f)             writer.writerow(('图片名', '电表读数', '电表编号'))             writer.writerows(zip(list_picturename, list_dbds, list_dbbh))         self.ms_signals_err_ocr.signal_err_ocr.emit("电表检测完成!")     # 点击重置按钮  触发 的事件 ，使得两个文本框 为空     def onclick2(self):         self.lineEdit.setText("")         self.lineEdit_2.setText("")         self.lineEdit_3.setText("")     # 点击第一个按钮 选择文件夹  触发的事件     def onclick3(self, filepath):         m = QtWidgets.QFileDialog.getExistingDirectory(None, "选取文件夹", "C:/")  # 起始路径         self.ms_signals_xz1.signal_xz1.emit(m)     # 点击第二个按钮 选择文件夹 触发 的事件     def onclick4(self, filepath):         m = QtWidgets.QFileDialog.getExistingDirectory(None, "选取文件夹", "C:/")  # 起始路径         self.ms_signals_xz2.signal_xz2.emit(m)     def init_ui(self):         window.Ui_Form.setupUi(self, self)         self.lineEdit.setPlaceholderText("请输入图片文件夹全路径（例如：C:\images）")         self.lineEdit.setText("请输入图片文件夹全路径（例如：C:\images）")         self.lineEdit_2.setPlaceholderText("请输入csv文件夹全路径（例如：C:\desktop\csv）")         self.lineEdit_2.setText("请输入csv文件夹全路径（例如：C:\desktop\csv）")         # 开始检测按钮         self.pushButton_3.clicked.connect(self.onclick1)         # 重置按钮         self.pushButton_4.clicked.connect(self.onclick2)         # 第一个选择文件夹 按钮         self.pushButton.clicked.connect(self.onclick3)         # 第二个选择文件夹 按钮         self.pushButton_2.clicked.connect(self.onclick4) if __name__ == '__main__':     app = QApplication(sys.argv)     window_ = Signals()     window_.show()     sys.exit(app.exec_())