毕设新项目基于PyQt5+PaddleOCR实现文字识别python源码.zip

from PyQt5 import QtGui from PyQt5 import QtCore from PyQt5 import QtWidgets from PyQt5.QtWidgets import QMainWindow, QListWidget, QListWidgetItem, QAbstractItemView, QWidget, QApplication, \ QButtonGroup, QPushButton, QTextEdit, QRadioButton, QCheckBox, QLabel, QSpacerItem, QMessageBox, QGroupBox, \ QVBoxLayout, QHBoxLayout from PyQt5.QtCore import QObject, QThread, QSettings, pyqtSignal, pyqtSlot, Qt from .logger import logger from .shape import Shape import PIL.Image import math import os import io import json import functools import imgviz from guiocr import __appname__ from guiocr import PY2 from guiocr import QT5 from guiocr import utils from guiocr.config import get_config from guiocr.widgets.main_window_ui import Ui_MainWindow from guiocr.widgets import * from guiocr.utils import * LABEL_COLORMAP = imgviz.label_colormap(value=200) here = os.path.dirname(os.path.abspath(__file__)) class MainWindow(QMainWindow): FIT_WINDOW, FIT_WIDTH, MANUAL_ZOOM = 0, 1, 2 def __init__(self): super().__init__() # 调用父类构造函数，创建QWidget窗体 self.last_selectBtnName = "" self.last_ComboxText = "" self._ui = Ui_MainWindow() # 创建ui对象 self._ui.setupUi(self) # 构造ui self.setWindowTitle(__appname__) # 加载默认配置 config = get_config() self._config = config # 程序数据 self.image = QtGui.QImage() self.dataDict = {} # 用于保持标注数据 self.imagePath = None self.recentFiles = [] self.maxRecent = 7 self.otherData = None self.zoom_level = 100 self.fit_window = False self.zoom_values = {} # key=filename, value=(zoom_mode, zoom_value) self.brightnessContrast_values = {} self.filename = "" self.output_dir = "./output" self.lastOpenDir = here self.imageList = [] self.result = [] self.suffix = ".json" self.scroll_values = { Qt.Horizontal: {}, Qt.Vertical: {}, } # key=filename, value=scroll_value # 线程 self.workThread = QThread() self.processor = OCR_qt() self.processor.moveToThread(self.workThread) self.processor.sendResult.connect(self.onReceiveResults) self.workThread.started.connect(self.processor.start) # 单选按钮组 self.checkBtnGroup = QButtonGroup(self) self.checkBtnGroup.addButton(self._ui.checkBox_ocr) # self.checkBtnGroup.addButton(self._ui.checkBox_det) # self.checkBtnGroup.addButton(self._ui.checkBox_recog) # self.checkBtnGroup.addButton(self._ui.checkBox_layoutparser) self.checkBtnGroup.setExclusive(True) # 添加按钮icon self._ui.btnOpenImg.setIcon(self.getIcon("open_img_grey")) self._ui.btnOpenDir.setIcon(self.getIcon("folder_open_grey")) self._ui.btnNext.setIcon(self.getIcon("next_grey")) self._ui.btnPrev.setIcon(self.getIcon("before_grey")) # self._ui.btnAddShape.setIcon(self.getIcon("add_grey")) # self._ui.btnEditShape.setIcon(self.getIcon("edit_grey")) self._ui.btnSaveAll.setIcon(self.getIcon("done_grey")) # self._ui.btnBrightness.setIcon(self.getIcon("brightness_grey")) # self._ui.btnStartProcess.setIcon(self.getIcon("play_white")) # 按钮响应函数 self._ui.btnOpenImg.clicked.connect(self.openFile) self._ui.btnOpenDir.clicked.connect(self.openDirDialog) self._ui.btnNext.clicked.connect(self.openNextImg) self._ui.btnPrev.clicked.connect(self.openPrevImg) self._ui.btnStartProcess.clicked.connect(self.startProcess) self._ui.btnCopyAll.clicked.connect(self.copyToClipboard) self._ui.btnSaveAll.clicked.connect(self.saveToFile) # self._ui.btnAddShape.clicked.connect(self.newShape) # self._ui.btnEditShape.clicked.connect(self.setEditMode) self._ui.listWidgetResults.itemClicked.connect(self.onItemResultClicked) # self._ui.listWidgetResults.itemSelectionChanged.connect(self.onItemResultClicked) self._ui.listWidgetResults.clear() # self.addResultItem(shape=None,txt="test3") # 控件布局 """左侧：区域标签列表""" self.labelList = LabelListWidget() self._ui.scrollAreaLabellist.setWidget(self.labelList) self._ui.scrollAreaLabellist.setWidgetResizable(True) self.labelList.itemSelectionChanged.connect(self.labelSelectionChanged) # self.labelList.itemDoubleClicked.connect(self.editLabel) self.labelList.itemChanged.connect(self.labelItemChanged) self.labelList.itemDropped.connect(self.labelOrderChanged) self.labelList.setSelectionMode(QAbstractItemView.MultiSelection) # 设置单选or多选 """缩放控件""" self.zoomWidget = ZoomWidget() self.setAcceptDrops(True) """中心画布""" self.canvas = self.labelList.canvas = Canvas( epsilon=self._config["epsilon"], double_click=self._config["canvas"]["double_click"], num_backups=self._config["canvas"]["num_backups"], ) self.canvas.zoomRequest.connect(self.zoomRequest) self.canvas.newShape.connect(self.newShape) self.canvas.shapeMoved.connect(self.onMoveShape) # self.setDirty) self.canvas.selectionChanged.connect(self.shapeSelectionChanged) self.canvas.drawingPolygon.connect(self.toggleDrawingSensitive) # Callbacks: self.zoomWidget.valueChanged.connect(self.paintCanvas) # 滚动缩放区域 self._ui.scrollAreaCanvas.setWidget(self.canvas) self._ui.scrollAreaCanvas.setWidgetResizable(True) self.scrollBars = { Qt.Vertical: self._ui.scrollAreaCanvas.verticalScrollBar(), Qt.Horizontal: self._ui.scrollAreaCanvas.horizontalScrollBar(), } self.canvas.scrollRequest.connect(self.scrollRequest) # self.setCentralWidget(self._ui.scrollAreaCanvas) # 设置默认形状颜色 Shape.line_color = QtGui.QColor(*self._config["shape"]["line_color"]) Shape.fill_color = QtGui.QColor(*self._config["shape"]["fill_color"]) Shape.select_line_color = QtGui.QColor( *self._config["shape"]["select_line_color"] ) Shape.select_fill_color = QtGui.QColor( *self._config["shape"]["select_fill_color"] ) Shape.vertex_fill_color = QtGui.QColor( *self._config["shape"]["vertex_fill_color"] ) Shape.hvertex_fill_color = QtGui.QColor( *self._config["shape"]["hvertex_fill_color"] ) # Restore application settings. self.settings = QSettings("app", __appname__) # actions self._initActions() # status bar self.statusBar().showMessage(str(self.tr("%s started.")) % __appname__) self.statusBar().show() # TODO 快捷键设置 def _initActions(self): # Actions action = functools.partial(utils.newAction, self) shortcuts = self._config["shortcuts"] quit = action( self.tr("&Quit"), self.close, shortcuts["quit"], "quit", self.tr("Quit application"), ) open_ = action( self.tr("&Open"), self.openFile, shortcuts["open"], "open", self.tr("Open image or label file"), ) opendir = action( self.tr("&Open Dir"), self.openDirDialog, shortcuts["open_dir"], "open", self.tr(u"Open Dir"), ) openNextImg = action( self.tr("&Next Image"), self.openNextImg, shortcuts["open_next"], "next", self.tr(u"Open next (hold Ctl+Shift to copy labels)"),