使用 Pyinstaller 打包 PPOCRLabel
使用 Pyinstaller 打包 PPOCRLabel
注意： 本文写于 2024-11-27
Clone 仓库与准备 Conda 环境
Fork PPOCRLabel 的官方仓库，然后 Clone
git clone git@github.com:shizidushu/PPOCRLabel.git
创建 conda 环境
conda create -n ppocrlabel python=3.10
conda activate ppocrlabel
python -m pip install paddlepaddle-gpu==2.6.1.post120 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
pip install git+https://gitee.com/paddlepaddle/PaddleOCR.git@v2.9.1
pyinstaller
安装依赖
pip install pyinstaller
使用命令查看 pyinstaller 的使用帮助文本（或者查看官网： https://pyinstaller.org/en/stable/usage.html ）：
pyinstaller --help
-p DIR, --paths DIR- A path to search for imports (like using PYTHONPATH). Multiple paths are allowed, separated by
':', or use this option multiple times. Equivalent to supplying thepathexargument in the spec file. - 从哪些文件夹路径搜寻模块，可能需要配合
--add-data一起使用
- A path to search for imports (like using PYTHONPATH). Multiple paths are allowed, separated by
--add-data SOURCE:DEST- Additional data files or directories containing data files to be added to the application. The argument value should be in form of “source:dest_dir”, where source is the path to file (or directory) to be collected, dest_dir is the destination directory relative to the top-level application directory, and both paths are separated by a colon (😃. To put a file in the top-level application directory, use . as a dest_dir. This option can be used multiple times.
- 添加数据文件
--add-binary SOURCE:DEST- Additional binary files to be added to the executable. See the
--add-dataoption for the format. This option can be used multiple times. - 添加 DLL 或者 SO 文件
- Additional binary files to be added to the executable. See the
--hidden-import MODULENAME, --hiddenimport MODULENAME- Name an import not visible in the code of the script(s). This option can be used multiple times.
--collect-all MODULENAME- Collect all submodules, data files, and binaries from the specified package or module. This option can be used multiple times
--clean- Clean PyInstaller cache and remove temporary files before building.
-w, --windowed, --noconsole- Windows and Mac OS X: do not provide a console window for standard i/o. On Mac OS this also triggers building a Mac OS .app bundle. On Windows this option is automatically set if the first script is a ‘.pyw’ file. This option is ignored on
*NIXsystems. - 比如一个 QT 应用，双击后会同时打开窗口和命令行，加上
-w之后命令行就消失了
- Windows and Mac OS X: do not provide a console window for standard i/o. On Mac OS this also triggers building a Mac OS .app bundle. On Windows this option is automatically set if the first script is a ‘.pyw’ file. This option is ignored on
初次打包
执行打包命令
# 重新生成资源pyrcc5 -o libs/resources.py resources.qrc
# 打包可执行程序pyinstaller -c PPOCRLabel.py --collect-all paddleocr --collect-all pyclipper --collect-all imghdr --collect-all skimage --collect-all imgaug --collect-all scipy.io --collect-all lmdb --hidden-import requests --collect-all paddle --collect-submodules shapely --hidden-import=pyqt5 -p ./libs --add-data ./libs:./libs -p ./ -p ./data --add-data ./data:./data -p ./resources --add-data ./resources:./resources --clean -F
修改代码后打包
修复 PPOCRLabel 显示 BOX 序号时的 BUG
修复显示 BOX 序号时的 BUG，修改 libs\shape.py 的 186 行为（使用 int()）：
painter.drawText(int(min_x), int(min_y), self.label)
修复 PPOCRLabel 会处理已 check 图片的 BUG
注释掉原先的 uncheckedList，使用新的逻辑：
# uncheckedList = [i for i in self.mImgList if i not in self.fileStatedict.keys()]checkedFilePaths = [os.path.join(*Path(file).parts[-2:]) for file in self.fileStatedict if self.fileStatedict[file] == 1]uncheckedList = [file for file in self.mImgList if os.path.join(*Path(file).parts[-2:]) not in checkedFilePaths]
添加更多 PaddleOCR 的参数
修改 MainWindow 的 __init__ 的函数签名，添加以下 3 个参数：
det_limit_type=None,
det_limit_side_len=None,
rec_image_shape=None,
修改函数主体，在 self.ocr = PaddleOCR(**params) 之前添加：
if det_limit_type is not None:params["det_limit_type"] = det_limit_typeif det_limit_side_len is not None:params["det_limit_side_len"] = det_limit_side_lenif rec_image_shape is not None:params["rec_image_shape"] = rec_image_shape
在 get_main_app 中添加：
arg_parser.add_argument("--det_limit_type", type=str, default=None, nargs="?")
arg_parser.add_argument("--det_limit_side_len", type=float, default=None, nargs="?")
arg_parser.add_argument("--rec_image_shape", type=str, default=None, nargs="?")
修改默认模型
设置默认模型
添加函数，方便使用相对路径：
def resolve_path(path):if getattr(sys, "frozen", False):# If the 'frozen' flag is set, we are in bundled-app mode!resolved_path = os.path.abspath(os.path.join(sys._MEIPASS, path))else:# Normal development mode. Use os.getcwd() or __file__ as appropriate in your case...resolved_path = os.path.abspath(os.path.join(os.getcwd(), path))return resolved_path
修改 get_main_app 函数，在 args = arg_parser.parse_args(argv[1:]) 之前添加：
arg_parser.set_defaults(det_model_dir=resolve_path("./custom_models/paddleocr-handwritten-digits/det/ch_PP-OCRv4_det_teacher_digits_demo_960_192"),det_limit_type="min",det_limit_side_len=192,rec_model_dir=resolve_path("./custom_models/paddleocr-handwritten-digits/rec/en_PP-OCRv3_rec_digits_demo"),rec_image_shape="3, 48, 48",rec_char_dict_path=resolve_path("./custom_models/paddleocr-handwritten-digits/rec/digits_dict.txt")# use_gpu=False)
修改 get_main_app 函数中调用 MainWindow 的代码：
win = MainWindow(lang=args.lang,gpu=args.gpu,img_list_natural_sort=args.img_list_natural_sort,kie_mode=args.kie,default_predefined_class_file=args.predefined_classes_file,det_model_dir=args.det_model_dir,det_limit_type=args.det_limit_type,det_limit_side_len=args.det_limit_side_len,rec_model_dir=args.rec_model_dir,rec_image_shape=args.rec_image_shape,rec_char_dict_path=args.rec_char_dict_path,cls_model_dir=args.cls_model_dir,bbox_auto_zoom_center=args.bbox_auto_zoom_center,label_font_path=args.label_font_path,selected_shape_color=args.selected_shape_color,)
根据模型需要，修改推理逻辑
修改 predict_system.py
此处采用的是自定义的手写数字识别模型，该模型需要修改 paddleocr 的 paddleocr\tools\infer\predict_system.py 文件：
# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import sys
import subprocess__dir__ = os.path.dirname(os.path.abspath(__file__))
sys.path.append(__dir__)
sys.path.insert(0, os.path.abspath(os.path.join(__dir__, "../..")))os.environ["FLAGS_allocator_strategy"] = "auto_growth"import cv2
import copy
import numpy as np
import json
import time
import logging
from PIL import Image
import tools.infer.utility as utility
import tools.infer.predict_rec as predict_rec
import tools.infer.predict_det as predict_det
import tools.infer.predict_cls as predict_cls
from ppocr.utils.utility import get_image_file_list, check_and_read
from ppocr.utils.logging import get_logger
from tools.infer.utility import (draw_ocr_box_txt,get_rotate_crop_image,get_minarea_rect_crop,slice_generator,merge_fragmented,
)logger = get_logger()class TextSystem(object):def __init__(self, args):if not args.show_log:logger.setLevel(logging.INFO)self.text_detector = predict_det.TextDetector(args)self.text_recognizer = predict_rec.TextRecognizer(args)self.use_angle_cls = args.use_angle_clsself.drop_score = args.drop_scoreif self.use_angle_cls:self.text_classifier = predict_cls.TextClassifier(args)self.args = argsself.crop_image_res_index = 0def draw_crop_rec_res(self, output_dir, img_crop_list, rec_res):os.makedirs(output_dir, exist_ok=True)bbox_num = len(img_crop_list)for bno in range(bbox_num):cv2.imwrite(os.path.join(output_dir, f"mg_crop_{bno+self.crop_image_res_index}.jpg"),img_crop_list[bno],)logger.debug(f"{bno}, {rec_res[bno]}")self.crop_image_res_index += bbox_numdef __call__(self, img, cls=True, slice={}):time_dict = {"det": 0, "rec": 0, "cls": 0, "all": 0}if img is None:logger.debug("no valid image provided")return None, None, time_dictstart = time.time()ori_im = img.copy()if slice:slice_gen = slice_generator(img,horizontal_stride=slice["horizontal_stride"],vertical_stride=slice["vertical_stride"],)elapsed = []dt_slice_boxes = []for slice_crop, v_start, h_start in slice_gen:dt_boxes, elapse = self.text_detector(slice_crop, use_slice=True)if dt_boxes.size:dt_boxes[:, :, 0] += h_startdt_boxes[:, :, 1] += v_startdt_slice_boxes.append(dt_boxes)elapsed.append(elapse)dt_boxes = np.concatenate(dt_slice_boxes)dt_boxes = merge_fragmented(boxes=dt_boxes,x_threshold=slice["merge_x_thres"],y_threshold=slice["merge_y_thres"],)elapse = sum(elapsed)else:dt_boxes, elapse = self.text_detector(img)time_dict["det"] = elapseif dt_boxes is None:logger.debug("no dt_boxes found, elapsed : {}".format(elapse))end = time.time()time_dict["all"] = end - startreturn None, None, time_dictelse:logger.debug("dt_boxes num : {}, elapsed : {}".format(len(dt_boxes), elapse))img_crop_list = []dt_boxes = sorted_boxes(dt_boxes)dt_boxes = sorted(dt_boxes, key=lambda x: x[0][0])new_dt_boxes = []for bno in range(len(dt_boxes)):tmp_box = copy.deepcopy(dt_boxes[bno])# print(tmp_box.dtype)# print(f'tmp_box: {tmp_box}')# if self.args.det_box_type == "quad":# img_crop = get_rotate_crop_image(ori_im, tmp_box)# else:# img_crop = get_minarea_rect_crop(ori_im, tmp_box)pt0, pt1, pt2, pt3 = tmp_box.astype(int).tolist()box_width = pt2[1] - pt0[1]box_height = pt2[0] - pt0[0]if box_width > 0 and box_height > 0 and (box_width > 2 and box_height > 2) and (box_width > 10 and box_height > 10):new_dt_boxes.append(np.array([pt0,[pt2[0], pt0[1]],pt2,[pt0[0], pt2[1]]], dtype=np.float32))img_crop = ori_im[pt0[1]:pt2[1], pt0[0]:pt2[0]]img_crop_list.append(img_crop)dt_boxes = new_dt_boxesif self.use_angle_cls and cls:img_crop_list, angle_list, elapse = self.text_classifier(img_crop_list)time_dict["cls"] = elapselogger.debug("cls num : {}, elapsed : {}".format(len(img_crop_list), elapse))if len(img_crop_list) > 1000:logger.debug(f"rec crops num: {len(img_crop_list)}, time and memory cost may be large.")# for img in img_crop_list:# img_crop_list = [img for img in img_crop_list if (img.shape[0] > 0 and img.shape[1] > 0) and (img.shape[0] > 2 and img.shape[1] > 2) and (img.shape[0] > 10 or img.shape[1] > 10)]rec_res, elapse = self.text_recognizer(img_crop_list)time_dict["rec"] = elapselogger.debug("rec_res num : {}, elapsed : {}".format(len(rec_res), elapse))if self.args.save_crop_res:self.draw_crop_rec_res(self.args.crop_res_save_dir, img_crop_list, rec_res)filter_boxes, filter_rec_res = [], []for box, rec_result in zip(dt_boxes, rec_res):text, score = rec_result[0], rec_result[1]if score >= self.drop_score:filter_boxes.append(box)filter_rec_res.append(rec_result)logger.debug(f"filter_boxes: {filter_boxes}\t filter_rec_res: {filter_rec_res}")end = time.time()time_dict["all"] = end - startreturn filter_boxes, filter_rec_res, time_dictdef sorted_boxes(dt_boxes):"""Sort text boxes in order from top to bottom, left to rightargs:dt_boxes(array):detected text boxes with shape [4, 2]return:sorted boxes(array) with shape [4, 2]"""num_boxes = dt_boxes.shape[0]sorted_boxes = sorted(dt_boxes, key=lambda x: (x[0][1], x[0][0]))_boxes = list(sorted_boxes)for i in range(num_boxes - 1):for j in range(i, -1, -1):if abs(_boxes[j + 1][0][1] - _boxes[j][0][1]) < 10 and (_boxes[j + 1][0][0] < _boxes[j][0][0]):tmp = _boxes[j]_boxes[j] = _boxes[j + 1]_boxes[j + 1] = tmpelse:breakreturn _boxesdef main(args):image_file_list = get_image_file_list(args.image_dir)image_file_list = image_file_list[args.process_id :: args.total_process_num]text_sys = TextSystem(args)is_visualize = Truefont_path = args.vis_font_pathdrop_score = args.drop_scoredraw_img_save_dir = args.draw_img_save_diros.makedirs(draw_img_save_dir, exist_ok=True)save_results = []logger.info("In PP-OCRv3, rec_image_shape parameter defaults to '3, 48, 320', ""if you are using recognition model with PP-OCRv2 or an older version, please set --rec_image_shape='3,32,320")# warm up 10 timesif args.warmup:img = np.random.uniform(0, 255, [640, 640, 3]).astype(np.uint8)for i in range(10):res = text_sys(img)total_time = 0cpu_mem, gpu_mem, gpu_util = 0, 0, 0_st = time.time()count = 0for idx, image_file in enumerate(image_file_list):logger.debug("loading image:{}".format(image_file))img, flag_gif, flag_pdf = check_and_read(image_file)if not flag_gif and not flag_pdf:img = cv2.imread(image_file)if not flag_pdf:if img is None:logger.debug("error in loading image:{}".format(image_file))continueimgs = [img]else:page_num = args.page_numif page_num > len(img) or page_num == 0:page_num = len(img)imgs = img[:page_num]for index, img in enumerate(imgs):starttime = time.time()dt_boxes, rec_res, time_dict = text_sys(img)elapse = time.time() - starttimetotal_time += elapseif len(imgs) > 1:logger.debug(str(idx)+ "_"+ str(index)+ " Predict time of %s: %.3fs" % (image_file, elapse))else:logger.debug(str(idx) + " Predict time of %s: %.3fs" % (image_file, elapse))for text, score in rec_res:logger.debug("{}, {:.3f}".format(text, score))res = [{"transcription": rec_res[i][0],"points": np.array(dt_boxes[i]).astype(np.int32).tolist(),}for i in range(len(dt_boxes))]if len(imgs) > 1:save_pred = (os.path.basename(image_file)+ "_"+ str(index)+ "\t"+ json.dumps(res, ensure_ascii=False)+ "\n")else:save_pred = (os.path.basename(image_file)+ "\t"+ json.dumps(res, ensure_ascii=False)+ "\n")save_results.append(save_pred)if is_visualize:image = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))boxes = dt_boxestxts = [rec_res[i][0] for i in range(len(rec_res))]scores = [rec_res[i][1] for i in range(len(rec_res))]draw_img = draw_ocr_box_txt(image,boxes,txts,scores,drop_score=drop_score,font_path=font_path,)if flag_gif:save_file = image_file[:-3] + "png"elif flag_pdf:save_file = image_file.replace(".pdf", "_" + str(index) + ".png")else:save_file = image_filecv2.imwrite(os.path.join(draw_img_save_dir, os.path.basename(save_file)),draw_img[:, :, ::-1],)logger.debug("The visualized image saved in {}".format(os.path.join(draw_img_save_dir, os.path.basename(save_file))))logger.info("The predict total time is {}".format(time.time() - _st))if args.benchmark:text_sys.text_detector.autolog.report()text_sys.text_recognizer.autolog.report()with open(os.path.join(draw_img_save_dir, "system_results.txt"), "w", encoding="utf-8") as f:f.writelines(save_results)if __name__ == "__main__":args = utility.parse_args()if args.use_mp:p_list = []total_process_num = args.total_process_numfor process_id in range(total_process_num):cmd = ([sys.executable, "-u"]+ sys.argv+ ["--process_id={}".format(process_id), "--use_mp={}".format(False)])p = subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stdout)p_list.append(p)for p in p_list:p.wait()else:main(args)
修改 PPOCRlabel.py 中的 reRecognition 函数
修改其中关于裁剪的部分：
# img_crop = get_rotate_crop_image(img, np.array(box, np.float32))# Do not rotatept0, pt1, pt2, pt3 = boximg_crop = img[int(pt0[1]):int(pt2[1]), int(pt0[0]):int(pt2[0])]
进行所有修改后的 PPOCRlabel.py 文件
# Copyright (c) <2015-Present> Tzutalin
# Copyright (C) 2013 MIT, Computer Science and Artificial Intelligence Laboratory. Bryan Russell, Antonio Torralba,
# William T. Freeman. Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
# associated documentation files (the "Software"), to deal in the Software without restriction, including without
# limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
# Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
# The above copyright notice and this permission notice shall be included in all copies or substantial portions of
# the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT
# NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
# SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
# CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
# !/usr/bin/env python
# -*- coding: utf-8 -*-
# pyrcc5 -o libs/resources.py resources.qrc
import argparse
import ast
import codecs
import json
import os
from pathlib import Path
import platform
import subprocess
import sys
import traceback
from functools import partialimport openpyxl
import cv2
import numpy as npfrom PyQt5.QtCore import (QSize,Qt,QPoint,QByteArray,QTimer,QFileInfo,QPointF,QProcess,
)
from PyQt5.QtGui import (QImage,QCursor,QPixmap,QImageReader,QColor,QIcon,QFontDatabase,
)
from PyQt5.QtWidgets import (QMainWindow,QListWidget,QVBoxLayout,QToolButton,QHBoxLayout,QDockWidget,QWidget,QSlider,QGraphicsOpacityEffect,QMessageBox,QListView,QScrollArea,QWidgetAction,QApplication,QLabel,QGridLayout,QFileDialog,QListWidgetItem,QComboBox,QDialog,QAbstractItemView,QMenu,QAction,QPushButton,
)__dir__ = os.path.dirname(__file__)
sys.path.append(os.path.join(__dir__, ""))from paddleocr import PaddleOCR, PPStructure
import libs.resources
from libs.constants import (SETTING_ADVANCE_MODE,SETTING_DRAW_SQUARE,SETTING_FILENAME,SETTING_FILL_COLOR,SETTING_LAST_OPEN_DIR,SETTING_LINE_COLOR,SETTING_PAINT_INDEX,SETTING_PAINT_LABEL,SETTING_RECENT_FILES,SETTING_SAVE_DIR,SETTING_WIN_POSE,SETTING_WIN_SIZE,SETTING_WIN_STATE,
)
from libs.utils import (addActions,boxPad,convert_token,expand_list,fmtShortcut,get_rotate_crop_image,have_qstring,keysInfo,natural_sort,newAction,newIcon,rebuild_html_from_ppstructure_label,stepsInfo,polygon_bounding_box_center_and_area,map_value,struct,
)
from libs.labelColor import label_colormap
from libs.settings import Settings
from libs.shape import Shape, DEFAULT_LINE_COLOR, DEFAULT_FILL_COLOR, DEFAULT_LOCK_COLOR
from libs.stringBundle import StringBundle
from libs.canvas import Canvas
from libs.zoomWidget import ZoomWidget
from libs.autoDialog import AutoDialog
from libs.labelDialog import LabelDialog
from libs.colorDialog import ColorDialog
from libs.ustr import ustr
from libs.hashableQListWidgetItem import HashableQListWidgetItem
from libs.editinlist import EditInList
from libs.unique_label_qlist_widget import UniqueLabelQListWidget
from libs.keyDialog import KeyDialog__appname__ = "PPOCRLabel"LABEL_COLORMAP = label_colormap()class MainWindow(QMainWindow):FIT_WINDOW, FIT_WIDTH, MANUAL_ZOOM = list(range(3))def __init__(self,lang="ch",gpu=False,img_list_natural_sort=True,bbox_auto_zoom_center=False,kie_mode=False,default_filename=None,default_predefined_class_file=None,default_save_dir=None,det_model_dir=None,det_limit_type=None,det_limit_side_len=None,rec_model_dir=None,rec_image_shape=None,rec_char_dict_path=None,cls_model_dir=None,label_font_path=None,selected_shape_color=(255, 255, 0),):super(MainWindow, self).__init__()self.setWindowTitle(__appname__)self.setWindowState(Qt.WindowMaximized) # set window maxself.activateWindow() # PPOCRLabel goes to the front when activate# Load setting in the main threadself.settings = Settings()self.settings.load()settings = self.settingsself.lang = langself.gpu = gpuself.img_list_natural_sort = img_list_natural_sortself.bbox_auto_zoom_center = bbox_auto_zoom_center# Load string bundle for i18nif lang not in ["ch", "en"]:lang = "en"self.stringBundle = StringBundle.getBundle(localeStr="zh-CN" if lang == "ch" else "en") # 'en'def getStr(strId):return self.stringBundle.getString(strId)# KIE settingself.kie_mode = kie_modeself.key_previous_text = ""self.existed_key_cls_set = set()self.key_dialog_tip = getStr("keyDialogTip")self.defaultSaveDir = default_save_dirparams = {"use_pdserving": False,"use_angle_cls": True,"det": True,"cls": True,"use_gpu": gpu,"lang": lang,"show_log": False,}if det_model_dir is not None:params["det_model_dir"] = det_model_dirif det_limit_type is not None:params["det_limit_type"] = det_limit_typeif det_limit_side_len is not None:params["det_limit_side_len"] = det_limit_side_lenif rec_model_dir is not None:params["rec_model_dir"] = rec_model_dirif rec_image_shape is not None:params["rec_image_shape"] = rec_image_shapeif rec_char_dict_path is not None:params["rec_char_dict_path"] = rec_char_dict_pathif cls_model_dir is not None:params["cls_model_dir"] = cls_model_dirprint(f"params.keys(): {params.keys()}")print(f"params: {params}")self.ocr = PaddleOCR(**params)self.table_ocr = PPStructure(use_pdserving=False, use_gpu=gpu, lang=lang, layout=False, show_log=False)if os.path.exists("./data/paddle.png"):result = self.ocr.ocr("./data/paddle.png", cls=True, det=True)result = self.table_ocr("./data/paddle.png", return_ocr_result_in_table=True)# For loading all image under a directoryself.mImgList = []self.mImgList5 = []self.dirname = Noneself.labelHist = []self.lastOpenDir = Noneself.result_dic = []self.result_dic_locked = []self.changeFileFolder = Falseself.haveAutoReced = Falseself.labelFile = Noneself.currIndex = 0# Whether we need to save or not.self.dirty = Falseself._noSelectionSlot = Falseself._beginner = Trueself.screencastViewer = self.getAvailableScreencastViewer()self.screencast = "https://github.com/PaddlePaddle/PaddleOCR"# Load predefined classes to the listself.loadPredefinedClasses(default_predefined_class_file)# Main widgets and related state.self.labelDialog = LabelDialog(parent=self, listItem=self.labelHist)self.autoDialog = AutoDialog(parent=self)self.itemsToShapes = {}self.shapesToItems = {}self.itemsToShapesbox = {}self.shapesToItemsbox = {}self.prevLabelText = getStr("tempLabel")self.noLabelText = getStr("nullLabel")self.model = "paddle"self.PPreader = Noneself.autoSaveNum = 5# ================== File List ==================filelistLayout = QVBoxLayout()filelistLayout.setContentsMargins(0, 0, 0, 0)self.fileListWidget = QListWidget()self.fileListWidget.itemClicked.connect(self.fileitemDoubleClicked)self.fileListWidget.setIconSize(QSize(25, 25))filelistLayout.addWidget(self.fileListWidget)fileListContainer = QWidget()fileListContainer.setLayout(filelistLayout)self.fileListName = getStr("fileList")self.fileDock = QDockWidget(self.fileListName, self)self.fileDock.setObjectName(getStr("files"))self.fileDock.setWidget(fileListContainer)self.addDockWidget(Qt.LeftDockWidgetArea, self.fileDock)# ================== Key List ==================if self.kie_mode:self.keyList = UniqueLabelQListWidget()# set key list heightkey_list_height = int(QApplication.desktop().height() // 4)if key_list_height < 50:key_list_height = 50self.keyList.setMaximumHeight(key_list_height)self.keyListDockName = getStr("keyListTitle")self.keyListDock = QDockWidget(self.keyListDockName, self)self.keyListDock.setWidget(self.keyList)self.keyListDock.setFeatures(QDockWidget.NoDockWidgetFeatures)filelistLayout.addWidget(self.keyListDock)self.AutoRecognition = QToolButton()self.AutoRecognition.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.AutoRecognition.setIcon(newIcon("Auto"))autoRecLayout = QHBoxLayout()autoRecLayout.setContentsMargins(0, 0, 0, 0)autoRecLayout.addWidget(self.AutoRecognition)autoRecContainer = QWidget()autoRecContainer.setLayout(autoRecLayout)filelistLayout.addWidget(autoRecContainer)# ================== Right Area ==================listLayout = QVBoxLayout()listLayout.setContentsMargins(0, 0, 0, 0)# Buttonsself.editButton = QToolButton()self.reRecogButton = QToolButton()self.reRecogButton.setIcon(newIcon("reRec", 30))self.reRecogButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.tableRecButton = QToolButton()self.tableRecButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.newButton = QToolButton()self.newButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.createpolyButton = QToolButton()self.createpolyButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.SaveButton = QToolButton()self.SaveButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.DelButton = QToolButton()self.DelButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)leftTopToolBox = QGridLayout()leftTopToolBox.addWidget(self.newButton, 0, 0, 1, 1)leftTopToolBox.addWidget(self.createpolyButton, 0, 1, 1, 1)leftTopToolBox.addWidget(self.reRecogButton, 1, 0, 1, 1)leftTopToolBox.addWidget(self.tableRecButton, 1, 1, 1, 1)leftTopToolBoxContainer = QWidget()leftTopToolBoxContainer.setLayout(leftTopToolBox)listLayout.addWidget(leftTopToolBoxContainer)# ================== Label List ==================labelIndexListlBox = QHBoxLayout()# Create and add a widget for showing current label item indexself.indexList = QListWidget()self.indexList.setMaximumSize(30, 16777215) # limit max widthself.indexList.setEditTriggers(QAbstractItemView.NoEditTriggers) # no editableself.indexList.itemSelectionChanged.connect(self.indexSelectionChanged)self.indexList.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff) # no scroll Barself.indexListDock = QDockWidget("No.", self)self.indexListDock.setWidget(self.indexList)self.indexListDock.setFeatures(QDockWidget.NoDockWidgetFeatures)labelIndexListlBox.addWidget(self.indexListDock, 1)# no margin between two boxeslabelIndexListlBox.setSpacing(0)# Create and add a widget for showing current label itemsself.labelList = EditInList()labelListContainer = QWidget()labelListContainer.setLayout(listLayout)self.labelList.itemSelectionChanged.connect(self.labelSelectionChanged)self.labelList.clicked.connect(self.labelList.item_clicked)# Connect to itemChanged to detect checkbox changes.self.labelList.itemChanged.connect(self.labelItemChanged)self.labelListDockName = getStr("recognitionResult")self.labelListDock = QDockWidget(self.labelListDockName, self)self.labelListDock.setWidget(self.labelList)self.labelListDock.setFeatures(QDockWidget.NoDockWidgetFeatures)labelIndexListlBox.addWidget(self.labelListDock, 10) # label list is wider than index list# enable labelList drag_drop to adjust bbox order# 设置选择模式为单选self.labelList.setSelectionMode(QAbstractItemView.SingleSelection)# 启用拖拽self.labelList.setDragEnabled(True)# 设置接受拖放self.labelList.viewport().setAcceptDrops(True)# 设置显示将要被放置的位置self.labelList.setDropIndicatorShown(True)# 设置拖放模式为移动项目，如果不设置，默认为复制项目self.labelList.setDragDropMode(QAbstractItemView.InternalMove)# 触发放置self.labelList.model().rowsMoved.connect(self.drag_drop_happened)labelIndexListContainer = QWidget()labelIndexListContainer.setLayout(labelIndexListlBox)listLayout.addWidget(labelIndexListContainer)# labelList indexList同步滚动self.labelListBar = self.labelList.verticalScrollBar()self.indexListBar = self.indexList.verticalScrollBar()self.labelListBar.valueChanged.connect(self.move_scrollbar)self.indexListBar.valueChanged.connect(self.move_scrollbar)# ================== Detection Box ==================self.BoxList = QListWidget()# self.BoxList.itemActivated.connect(self.boxSelectionChanged)self.BoxList.itemSelectionChanged.connect(self.boxSelectionChanged)self.BoxList.itemDoubleClicked.connect(self.editBox)# Connect to itemChanged to detect checkbox changes.self.BoxList.itemChanged.connect(self.boxItemChanged)self.BoxListDockName = getStr("detectionBoxposition")self.BoxListDock = QDockWidget(self.BoxListDockName, self)self.BoxListDock.setWidget(self.BoxList)self.BoxListDock.setFeatures(QDockWidget.NoDockWidgetFeatures)listLayout.addWidget(self.BoxListDock)# ================== Lower Right Area ==================leftbtmtoolbox = QHBoxLayout()leftbtmtoolbox.addWidget(self.SaveButton)leftbtmtoolbox.addWidget(self.DelButton)leftbtmtoolboxcontainer = QWidget()leftbtmtoolboxcontainer.setLayout(leftbtmtoolbox)listLayout.addWidget(leftbtmtoolboxcontainer)self.dock = QDockWidget(getStr("boxLabelText"), self)self.dock.setObjectName(getStr("labels"))self.dock.setWidget(labelListContainer)# ================== Zoom Bar ==================self.imageSlider = QSlider(Qt.Horizontal)self.imageSlider.valueChanged.connect(self.CanvasSizeChange)self.imageSlider.setMinimum(-9)self.imageSlider.setMaximum(510)self.imageSlider.setSingleStep(1)self.imageSlider.setTickPosition(QSlider.TicksBelow)self.imageSlider.setTickInterval(1)op = QGraphicsOpacityEffect()op.setOpacity(0.2)self.imageSlider.setGraphicsEffect(op)self.imageSlider.setStyleSheet("background-color:transparent")self.imageSliderDock = QDockWidget(getStr("ImageResize"), self)self.imageSliderDock.setObjectName(getStr("IR"))self.imageSliderDock.setWidget(self.imageSlider)self.imageSliderDock.setFeatures(QDockWidget.DockWidgetFloatable)self.imageSliderDock.setAttribute(Qt.WA_TranslucentBackground)self.addDockWidget(Qt.RightDockWidgetArea, self.imageSliderDock)self.zoomWidget = ZoomWidget()self.colorDialog = ColorDialog(parent=self)self.zoomWidgetValue = self.zoomWidget.value()self.msgBox = QMessageBox()# ================== Thumbnail ==================hlayout = QHBoxLayout()m = (0, 0, 0, 0)hlayout.setSpacing(0)hlayout.setContentsMargins(*m)self.preButton = QToolButton()self.preButton.setIcon(newIcon("prev", 40))self.preButton.setIconSize(QSize(40, 100))self.preButton.clicked.connect(self.openPrevImg)self.preButton.setStyleSheet("border: none;")self.preButton.setShortcut("a")self.iconlist = QListWidget()self.iconlist.setViewMode(QListView.IconMode)self.iconlist.setFlow(QListView.TopToBottom)self.iconlist.setSpacing(10)self.iconlist.setIconSize(QSize(50, 50))self.iconlist.setMovement(QListView.Static)self.iconlist.setResizeMode(QListView.Adjust)self.iconlist.itemClicked.connect(self.iconitemDoubleClicked)self.iconlist.setStyleSheet("QListWidget{ background-color:transparent; border: none;}")self.iconlist.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)self.nextButton = QToolButton()self.nextButton.setIcon(newIcon("next", 40))self.nextButton.setIconSize(QSize(40, 100))self.nextButton.setStyleSheet("border: none;")self.nextButton.clicked.connect(self.openNextImg)self.nextButton.setShortcut("d")hlayout.addWidget(self.preButton)hlayout.addWidget(self.iconlist)hlayout.addWidget(self.nextButton)iconListContainer = QWidget()iconListContainer.setLayout(hlayout)iconListContainer.setFixedHeight(100)# ================== Canvas ==================self.canvas = Canvas(parent=self)self.canvas.zoomRequest.connect(self.zoomRequest)self.canvas.setDrawingShapeToSquare(settings.get(SETTING_DRAW_SQUARE, False))scroll = QScrollArea()scroll.setWidget(self.canvas)scroll.setWidgetResizable(True)self.scrollBars = {Qt.Vertical: scroll.verticalScrollBar(),Qt.Horizontal: scroll.horizontalScrollBar(),}self.scrollArea = scrollself.canvas.scrollRequest.connect(self.scrollRequest)self.canvas.newShape.connect(partial(self.newShape, False))self.canvas.shapeMoved.connect(self.updateBoxlist) # self.setDirtyself.canvas.selectionChanged.connect(self.shapeSelectionChanged)self.canvas.drawingPolygon.connect(self.toggleDrawingSensitive)centerLayout = QVBoxLayout()centerLayout.setContentsMargins(0, 0, 0, 0)centerLayout.addWidget(scroll)centerLayout.addWidget(iconListContainer, 0, Qt.AlignCenter)centerContainer = QWidget()centerContainer.setLayout(centerLayout)self.setCentralWidget(centerContainer)self.addDockWidget(Qt.RightDockWidgetArea, self.dock)self.dock.setFeatures(QDockWidget.DockWidgetClosable | QDockWidget.DockWidgetFloatable)self.fileDock.setFeatures(QDockWidget.NoDockWidgetFeatures)# ================== Actions ==================action = partial(newAction, self)quit = action(getStr("quit"), self.close, "Ctrl+Q", "quit", getStr("quitApp"))opendir = action(getStr("openDir"), self.openDirDialog, "Ctrl+u", "open", getStr("openDir"))open_dataset_dir = action(getStr("openDatasetDir"),self.openDatasetDirDialog,"Ctrl+p","open",getStr("openDatasetDir"),enabled=False,)save = action(getStr("save"),self.saveFile,["Ctrl+V", "end"],"verify",getStr("saveDetail"),enabled=False,)alcm = action(getStr("choosemodel"),self.autolcm,"Ctrl+M","next",getStr("tipchoosemodel"),)deleteImg = action(getStr("deleteImg"),self.deleteImg,"Ctrl+Shift+D","close",getStr("deleteImgDetail"),enabled=True,)resetAll = action(getStr("resetAll"),self.resetAll,None,"resetall",getStr("resetAllDetail"),)color1 = action(getStr("boxLineColor"),self.chooseColor,"Ctrl+L","color_line",getStr("boxLineColorDetail"),)createMode = action(getStr("crtBox"),self.setCreateMode,"w","new",getStr("crtBoxDetail"),enabled=False,)editMode = action("&Edit\nRectBox",self.setEditMode,"Ctrl+J","edit","Move and edit Boxs",enabled=False,)create = action(getStr("crtBox"),self.createShape,"w","objects",getStr("crtBoxDetail"),enabled=False,)delete = action(getStr("delBox"),self.deleteSelectedShape,["backspace", "delete"],"delete",getStr("delBoxDetail"),enabled=False,)copy = action(getStr("dupBox"),self.copySelectedShape,"Ctrl+C","copy",getStr("dupBoxDetail"),enabled=False,)hideAll = action(getStr("hideBox"),partial(self.togglePolygons, False),"Ctrl+H","hide",getStr("hideAllBoxDetail"),enabled=False,)showAll = action(getStr("showBox"),partial(self.togglePolygons, True),"Ctrl+A","hide",getStr("showAllBoxDetail"),enabled=False,)help = action(getStr("tutorial"),self.showTutorialDialog,None,"help",getStr("tutorialDetail"),)showInfo = action(getStr("info"), self.showInfoDialog, None, "help", getStr("info"))showSteps = action(getStr("steps"), self.showStepsDialog, None, "help", getStr("steps"))showKeys = action(getStr("keys"), self.showKeysDialog, None, "help", getStr("keys"))zoom = QWidgetAction(self)zoom.setDefaultWidget(self.zoomWidget)self.zoomWidget.setWhatsThis("Zoom in or out of the image. Also accessible with"" %s and %s from the canvas."% (fmtShortcut("Ctrl+[-+]"), fmtShortcut("Ctrl+Wheel")))self.zoomWidget.setEnabled(False)zoomIn = action(getStr("zoomin"),partial(self.addZoom, 10),"Ctrl++","zoom-in",getStr("zoominDetail"),enabled=False,)zoomOut = action(getStr("zoomout"),partial(self.addZoom, -10),"Ctrl+-","zoom-out",getStr("zoomoutDetail"),enabled=False,)zoomOrg = action(getStr("originalsize"),partial(self.setZoom, 100),"Ctrl+=","zoom",getStr("originalsizeDetail"),enabled=False,)fitWindow = action(getStr("fitWin"),self.setFitWindow,"Ctrl+F","fit-window",getStr("fitWinDetail"),checkable=True,enabled=False,)fitWidth = action(getStr("fitWidth"),self.setFitWidth,"Ctrl+Shift+F","fit-width",getStr("fitWidthDetail"),checkable=True,enabled=False,)# Group zoom controls into a list for easier toggling.zoomActions = (self.zoomWidget, zoomIn, zoomOut, zoomOrg, fitWindow, fitWidth)self.zoomMode = self.MANUAL_ZOOMself.scalers = {self.FIT_WINDOW: self.scaleFitWindow,self.FIT_WIDTH: self.scaleFitWidth,# Set to one to scale to 100% when loading files.self.MANUAL_ZOOM: lambda: 1,}# ================== New Actions ==================edit = action(getStr("editLabel"),self.editLabel,"Ctrl+E","edit",getStr("editLabelDetail"),enabled=False,)AutoRec = action(getStr("autoRecognition"),self.autoRecognition,"","Auto",getStr("autoRecognition"),enabled=False,)reRec = action(getStr("reRecognition"),self.reRecognition,"Ctrl+Shift+R","reRec",getStr("reRecognition"),enabled=False,)singleRere = action(getStr("singleRe"),self.singleRerecognition,"Ctrl+R","reRec",getStr("singleRe"),enabled=False,)createpoly = action(getStr("creatPolygon"),self.createPolygon,["q", "home"],"new",getStr("creatPolygon"),enabled=False,)tableRec = action(getStr("TableRecognition"),self.TableRecognition,"","Auto",getStr("TableRecognition"),enabled=False,)cellreRec = action(getStr("cellreRecognition"),self.cellreRecognition,"","reRec",getStr("cellreRecognition"),enabled=False,)saveRec = action(getStr("saveRec"),self.saveRecResult,"","save",getStr("saveRec"),enabled=False,)saveLabel = action(getStr("saveLabel"),self.saveLabelFile, #"Ctrl+S","save",getStr("saveLabel"),enabled=False,)exportJSON = action(getStr("exportJSON"),self.exportJSON,"","save",getStr("exportJSON"),enabled=False,)undoLastPoint = action(getStr("undoLastPoint"),self.canvas.undoLastPoint,"Ctrl+Z","undo",getStr("undoLastPoint"),enabled=False,)rotateLeft = action(getStr("rotateLeft"),partial(self.rotateImgAction, 1),"Ctrl+Alt+L","rotateLeft",getStr("rotateLeft"),enabled=False,)rotateRight = action(getStr("rotateRight"),partial(self.rotateImgAction, -1),"Ctrl+Alt+R","rotateRight",getStr("rotateRight"),enabled=False,)undo = action(getStr("undo"),self.undoShapeEdit,"Ctrl+Z","undo",getStr("undo"),enabled=False,)change_cls = action(getStr("keyChange"),self.change_box_key,"Ctrl+X","edit",getStr("keyChange"),enabled=False,)lock = action(getStr("lockBox"),self.lockSelectedShape,None,"lock",getStr("lockBoxDetail"),enabled=False,)expand = action(getStr("expandBox"),self.expandSelectedShape,"Ctrl+K","expand",getStr("expandBoxDetail"),enabled=False,)self.editButton.setDefaultAction(edit)self.newButton.setDefaultAction(create)self.createpolyButton.setDefaultAction(createpoly)self.DelButton.setDefaultAction(deleteImg)self.SaveButton.setDefaultAction(save)self.AutoRecognition.setDefaultAction(AutoRec)self.reRecogButton.setDefaultAction(reRec)self.tableRecButton.setDefaultAction(tableRec)# self.preButton.setDefaultAction(openPrevImg)# self.nextButton.setDefaultAction(openNextImg)# ================== Zoom layout ==================zoomLayout = QHBoxLayout()zoomLayout.addStretch()self.zoominButton = QToolButton()self.zoominButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.zoominButton.setDefaultAction(zoomIn)self.zoomoutButton = QToolButton()self.zoomoutButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.zoomoutButton.setDefaultAction(zoomOut)self.zoomorgButton = QToolButton()self.zoomorgButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)self.zoomorgButton.setDefaultAction(zoomOrg)zoomLayout.addWidget(self.zoominButton)zoomLayout.addWidget(self.zoomorgButton)zoomLayout.addWidget(self.zoomoutButton)zoomContainer = QWidget()zoomContainer.setLayout(zoomLayout)zoomContainer.setGeometry(0, 0, 30, 150)shapeLineColor = action(getStr("shapeLineColor"),self.chshapeLineColor,icon="color_line",tip=getStr("shapeLineColorDetail"),enabled=False,)shapeFillColor = action(getStr("shapeFillColor"),self.chshapeFillColor,icon="color",tip=getStr("shapeFillColorDetail"),enabled=False,)# Label list context menu.labelMenu = QMenu()addActions(labelMenu, (edit, delete))self.labelList.setContextMenuPolicy(Qt.CustomContextMenu)self.labelList.customContextMenuRequested.connect(self.popLabelListMenu)# Draw squares/rectanglesself.drawSquaresOption = QAction(getStr("drawSquares"), self)self.drawSquaresOption.setCheckable(True)self.drawSquaresOption.setChecked(settings.get(SETTING_DRAW_SQUARE, False))self.drawSquaresOption.triggered.connect(self.toogleDrawSquare)# Store actions for further handling.self.actions = struct(save=save,resetAll=resetAll,deleteImg=deleteImg,lineColor=color1,create=create,createpoly=createpoly,tableRec=tableRec,delete=delete,edit=edit,copy=copy,saveRec=saveRec,singleRere=singleRere,AutoRec=AutoRec,reRec=reRec,cellreRec=cellreRec,createMode=createMode,editMode=editMode,shapeLineColor=shapeLineColor,shapeFillColor=shapeFillColor,zoom=zoom,zoomIn=zoomIn,zoomOut=zoomOut,zoomOrg=zoomOrg,fitWindow=fitWindow,fitWidth=fitWidth,zoomActions=zoomActions,saveLabel=saveLabel,change_cls=change_cls,undo=undo,undoLastPoint=undoLastPoint,open_dataset_dir=open_dataset_dir,rotateLeft=rotateLeft,rotateRight=rotateRight,lock=lock,exportJSON=exportJSON,expand=expand,fileMenuActions=(opendir,open_dataset_dir,saveLabel,exportJSON,resetAll,quit,),beginner=(),advanced=(),editMenu=(createpoly,edit,copy,delete,singleRere,cellreRec,None,undo,undoLastPoint,None,rotateLeft,rotateRight,None,color1,self.drawSquaresOption,lock,expand,None,change_cls,),beginnerContext=(create,createpoly,edit,copy,delete,singleRere,cellreRec,rotateLeft,rotateRight,lock,expand,change_cls,),advancedContext=(createMode,editMode,edit,copy,delete,shapeLineColor,shapeFillColor,),onLoadActive=(create, createpoly, createMode, editMode),onShapesPresent=(hideAll, showAll),)# menusself.menus = struct(file=self.menu("&" + getStr("mfile")),edit=self.menu("&" + getStr("medit")),view=self.menu("&" + getStr("mview")),autolabel=self.menu("&PaddleOCR"),help=self.menu("&" + getStr("mhelp")),recentFiles=QMenu("Open &Recent"),labelList=labelMenu,)self.lastLabel = None# Add option to enable/disable labels being displayed at the top of bounding boxesself.displayLabelOption = QAction(getStr("displayLabel"), self)self.displayLabelOption.setShortcut("Ctrl+Shift+P")self.displayLabelOption.setCheckable(True)self.displayLabelOption.setChecked(settings.get(SETTING_PAINT_LABEL, False))self.displayLabelOption.triggered.connect(self.togglePaintLabelsOption)# Add option to enable/disable box index being displayed at the top of bounding boxesself.displayIndexOption = QAction(getStr("displayIndex"), self)self.displayIndexOption.setCheckable(True)self.displayIndexOption.setChecked(settings.get(SETTING_PAINT_INDEX, False))self.displayIndexOption.triggered.connect(self.togglePaintIndexOption)self.labelDialogOption = QAction(getStr("labelDialogOption"), self)self.labelDialogOption.setShortcut("Ctrl+Shift+L")self.labelDialogOption.setCheckable(True)self.labelDialogOption.setChecked(settings.get(SETTING_PAINT_LABEL, False))self.displayIndexOption.setChecked(settings.get(SETTING_PAINT_INDEX, False))self.labelDialogOption.triggered.connect(self.speedChoose)self.autoSaveOption = QAction(getStr("autoSaveMode"), self)self.autoSaveOption.setCheckable(True)self.autoSaveOption.setChecked(settings.get(SETTING_PAINT_LABEL, False))self.displayIndexOption.setChecked(settings.get(SETTING_PAINT_INDEX, False))self.autoSaveOption.triggered.connect(self.autoSaveFunc)self.autoReRecognitionOption = QAction(getStr("autoReRecognition"), self)self.autoReRecognitionOption.setCheckable(True)self.autoReRecognitionOption.setChecked(settings.get(SETTING_PAINT_LABEL, False))self.displayIndexOption.setChecked(settings.get(SETTING_PAINT_INDEX, False))self.autoReRecognitionOption.triggered.connect(self.autoSaveFunc)self.autoSaveUnsavedChangesOption = QAction(getStr("autoSaveUnsavedChanges"), self)self.autoSaveUnsavedChangesOption.setCheckable(True)self.autoSaveUnsavedChangesOption.setChecked(settings.get(SETTING_PAINT_LABEL, False))self.displayIndexOption.setChecked(settings.get(SETTING_PAINT_INDEX, False))self.autoSaveUnsavedChangesOption.triggered.connect(self.autoSaveFunc)addActions(self.menus.file,(opendir,open_dataset_dir,None,saveLabel,saveRec,exportJSON,self.autoSaveOption,self.autoReRecognitionOption,self.autoSaveUnsavedChangesOption,None,resetAll,deleteImg,quit,),)addActions(self.menus.help, (showKeys, showSteps, showInfo))addActions(self.menus.view,(self.displayLabelOption,self.displayIndexOption,self.labelDialogOption,None,hideAll,showAll,None,zoomIn,zoomOut,zoomOrg,None,fitWindow,fitWidth,),)addActions(self.menus.autolabel, (AutoRec, reRec, cellreRec, alcm, None, help))self.menus.file.aboutToShow.connect(self.updateFileMenu)# Custom context menu for the canvas widget:addActions(self.canvas.menus[0], self.actions.beginnerContext)self.statusBar().showMessage("%s started." % __appname__)self.statusBar().show()# Application state.self.image = QImage()self.filePath = ustr(default_filename)self.lastOpenDir = Noneself.recentFiles = []self.maxRecent = 7self.lineColor = Noneself.fillColor = Noneself.zoom_level = 100self.fit_window = False# Add Chrisself.difficult = False# Fix the compatible issue for qt4 and qt5. Convert the QStringList to python listif settings.get(SETTING_RECENT_FILES):if have_qstring():recentFileQStringList = settings.get(SETTING_RECENT_FILES)self.recentFiles = [ustr(i) for i in recentFileQStringList]else:self.recentFiles = recentFileQStringList = settings.get(SETTING_RECENT_FILES)size = settings.get(SETTING_WIN_SIZE, QSize(1200, 800))position = QPoint(0, 0)saved_position = settings.get(SETTING_WIN_POSE, position)# Fix the multiple monitors issuefor i in range(QApplication.desktop().screenCount()):if QApplication.desktop().availableGeometry(i).contains(saved_position):position = saved_positionbreakself.resize(size)self.move(position)saveDir = ustr(settings.get(SETTING_SAVE_DIR, None))self.lastOpenDir = ustr(settings.get(SETTING_LAST_OPEN_DIR, None))self.restoreState(settings.get(SETTING_WIN_STATE, QByteArray()))Shape.line_color = self.lineColor = QColor(settings.get(SETTING_LINE_COLOR, DEFAULT_LINE_COLOR))Shape.fill_color = self.fillColor = QColor(settings.get(SETTING_FILL_COLOR, DEFAULT_FILL_COLOR))self.canvas.setDrawingColor(self.lineColor)# Add chrisShape.difficult = self.difficult# ADD:# Populate the File menu dynamically.self.updateFileMenu()# Since loading the file may take some time, make sure it runs in the background.if self.filePath and os.path.isdir(self.filePath):self.queueEvent(partial(self.importDirImages, self.filePath or ""))elif self.filePath:self.queueEvent(partial(self.loadFile, self.filePath or ""))self.keyDialog = None# Callbacks:self.zoomWidget.valueChanged.connect(self.paintCanvas)self.populateModeActions()# Display cursor coordinates at the right of status barself.labelCoordinates = QLabel("")self.statusBar().addPermanentWidget(self.labelCoordinates)# Open Dir if deafult fileif self.filePath and os.path.isdir(self.filePath):self.openDirDialog(dirpath=self.filePath, silent=True)# load label fontself.label_font_family = Noneif label_font_path is not None:label_font_id = QFontDatabase.addApplicationFont(label_font_path)if label_font_id >= 0:self.label_font_family = QFontDatabase.applicationFontFamilies(label_font_id)[0]# selected shape colorself.selected_shape_color = selected_shape_colordef menu(self, title, actions=None):menu = self.menuBar().addMenu(title)if actions:addActions(menu, actions)return menudef keyReleaseEvent(self, event):if event.key() == Qt.Key_Control:self.canvas.setDrawingShapeToSquare(False)def keyPressEvent(self, event):if event.key() == Qt.Key_Control:# Draw rectangle if Ctrl is pressedself.canvas.setDrawingShapeToSquare(True)def noShapes(self):return not self.itemsToShapesdef populateModeActions(self):self.canvas.menus[0].clear()addActions(self.canvas.menus[0], self.actions.beginnerContext)self.menus.edit.clear()actions = (self.actions.create,) # if self.beginner() else (self.actions.createMode, self.actions.editMode)addActions(self.menus.edit, actions + self.actions.editMenu)def setDirty(self):self.dirty = Trueself.actions.save.setEnabled(True)def setClean(self):self.dirty = Falseself.actions.save.setEnabled(False)self.actions.create.setEnabled(True)self.actions.createpoly.setEnabled(True)def toggleActions(self, value=True):"""Enable/Disable widgets which depend on an opened image."""for z in self.actions.zoomActions:z.setEnabled(value)for action in self.actions.onLoadActive:action.setEnabled(value)def queueEvent(self, function):QTimer.singleShot(0, function)def status(self, message, delay=5000):self.statusBar().showMessage(message, delay)def resetState(self):self.itemsToShapes.clear()self.shapesToItems.clear()self.itemsToShapesbox.clear() # ADDself.shapesToItemsbox.clear()self.labelList.clear()self.BoxList.clear()self.indexList.clear()self.filePath = Noneself.imageData = Noneself.labelFile = Noneself.canvas.resetState()self.labelCoordinates.clear()# self.comboBox.cb.clear()self.result_dic = []def currentItem(self):items = self.labelList.selectedItems()if items:return items[0]return Nonedef currentBox(self):items = self.BoxList.selectedItems()if items:return items[0]return Nonedef addRecentFile(self, filePath):if filePath in self.recentFiles:self.recentFiles.remove(filePath)elif len(self.recentFiles) >= self.maxRecent:self.recentFiles.pop()self.recentFiles.insert(0, filePath)def beginner(self):return self._beginnerdef advanced(self):return not self.beginner()def getAvailableScreencastViewer(self):osName = platform.system()if osName == "Windows":return ["C:\\Program Files\\Internet Explorer\\iexplore.exe"]elif osName == "Linux":return ["xdg-open"]elif osName == "Darwin":return ["open"]## Callbacks ##def showTutorialDialog(self):subprocess.Popen(self.screencastViewer + [self.screencast])def showInfoDialog(self):from libs.__init__ import __version__msg = "Name:{0} \nApp Version:{1} \n{2} ".format(__appname__, __version__, sys.version_info)QMessageBox.information(self, "Information", msg)def showStepsDialog(self):msg = stepsInfo(self.lang)QMessageBox.information(self, "Information", msg)def showKeysDialog(self):msg = keysInfo(self.lang)QMessageBox.information(self, "Information", msg)def createShape(self):assert self.beginner()self.canvas.setEditing(False)self.actions.create.setEnabled(False)self.actions.createpoly.setEnabled(False)self.canvas.fourpoint = Falsedef createPolygon(self):assert self.beginner()self.canvas.setEditing(False)self.canvas.fourpoint = Trueself.actions.create.setEnabled(False)self.actions.createpoly.setEnabled(False)self.actions.undoLastPoint.setEnabled(True)def rotateImg(self, filename, k, _value):self.actions.rotateRight.setEnabled(_value)pix = cv2.imdecode(np.fromfile(filename, dtype=np.uint8), cv2.IMREAD_COLOR)pix = np.rot90(pix, k)ext = os.path.splitext(filename)[1]cv2.imencode(ext, pix)[1].tofile(filename)self.canvas.update()self.loadFile(filename)def rotateImgWarn(self):if self.lang == "ch":self.msgBox.warning(self,"提示","\n 该图片已经有标注框,旋转操作会打乱标注,建议清除标注框后旋转。",)else:self.msgBox.warning(self,"Warn","\n The picture already has a label box, ""and rotation will disrupt the label. ""It is recommended to clear the label box and rotate it.",)def rotateImgAction(self, k=1, _value=False):filename = self.filePathif os.path.exists(filename):if self.itemsToShapesbox:self.rotateImgWarn()else:self.saveFile()self.dirty = Falseself.rotateImg(filename=filename, k=k, _value=True)else:self.rotateImgWarn()self.actions.rotateRight.setEnabled(False)self.actions.rotateLeft.setEnabled(False)def toggleDrawingSensitive(self, drawing=True):"""In the middle of drawing, toggling between modes should be disabled."""self.actions.editMode.setEnabled(not drawing)if not drawing and self.beginner():# Cancel creation.print("Cancel creation.")self.canvas.setEditing(True)self.canvas.restoreCursor()self.actions.create.setEnabled(True)self.actions.createpoly.setEnabled(True)def toggleDrawMode(self, edit=True):self.canvas.setEditing(edit)self.actions.createMode.setEnabled(edit)self.actions.editMode.setEnabled(not edit)def setCreateMode(self):assert self.advanced()self.toggleDrawMode(False)def setEditMode(self):assert self.advanced()self.toggleDrawMode(True)self.labelSelectionChanged()def updateFileMenu(self):currFilePath = self.filePathdef exists(filename):return os.path.exists(filename)menu = self.menus.recentFilesmenu.clear()files = [f for f in self.recentFiles if f != currFilePath and exists(f)]for i, f in enumerate(files):icon = newIcon("labels")action = QAction(icon, "&%d %s" % (i + 1, QFileInfo(f).fileName()), self)action.triggered.connect(partial(self.loadRecent, f))menu.addAction(action)def popLabelListMenu(self, point):self.menus.labelList.exec_(self.labelList.mapToGlobal(point))def editLabel(self):if not self.canvas.editing():returnitem = self.currentItem()if not item:returntext = self.labelDialog.popUp(item.text())if text is not None:item.setText(text)# item.setBackground(generateColorByText(text))self.setDirty()self.updateComboBox()# =================== detection box related functions ===================def boxItemChanged(self, item):shape = self.itemsToShapesbox[item]box = ast.literal_eval(item.text())# print('shape in labelItemChanged is',shape.points)if box != [(int(p.x()), int(p.y())) for p in shape.points]:# shape.points = boxshape.points = [QPointF(p[0], p[1]) for p in box]# QPointF(x,y)# shape.line_color = generateColorByText(shape.label)self.setDirty()else: # User probably changed item visibilityself.canvas.setShapeVisible(shape, True) # item.checkState() == Qt.Checkeddef editBox(self): # ADDif not self.canvas.editing():returnitem = self.currentBox()if not item:returntext = self.labelDialog.popUp(item.text())imageSize = str(self.image.size())width, height = self.image.width(), self.image.height()if text:try:text_list = eval(text)except Exception:msg_box = QMessageBox(QMessageBox.Warning, "Warning", "Please enter the correct format")msg_box.exec_()returnif len(text_list) < 4:msg_box = QMessageBox(QMessageBox.Warning,"Warning","Please enter the coordinates of 4 points",)msg_box.exec_()returnfor box in text_list:if box[0] > width or box[0] < 0 or box[1] > height or box[1] < 0:msg_box = QMessageBox(QMessageBox.Warning, "Warning", "Out of picture size")msg_box.exec_()returnitem.setText(text)# item.setBackground(generateColorByText(text))self.setDirty()self.updateComboBox()def updateBoxlist(self):self.canvas.selectedShapes_hShape = []if self.canvas.hShape is not None:self.canvas.selectedShapes_hShape = self.canvas.selectedShapes + [self.canvas.hShape]else:self.canvas.selectedShapes_hShape = self.canvas.selectedShapesfor shape in self.canvas.selectedShapes_hShape:if shape in self.shapesToItemsbox.keys():item = self.shapesToItemsbox[shape] # listitemtext = [(int(p.x()), int(p.y())) for p in shape.points]item.setText(str(text))self.actions.undo.setEnabled(True)self.setDirty()def indexTo5Files(self, currIndex):if currIndex < 2:return self.mImgList[:5]elif currIndex > len(self.mImgList) - 3:return self.mImgList[-5:]else:return self.mImgList[currIndex - 2 : currIndex + 3]# Tzutalin 20160906 : Add file list and dock to move fasterdef fileitemDoubleClicked(self, item=None):self.currIndex = self.mImgList.index(ustr(os.path.join(os.path.abspath(self.dirname), item.text())))filename = self.mImgList[self.currIndex]if filename:self.mImgList5 = self.indexTo5Files(self.currIndex)# self.additems5(None)self.loadFile(filename)def iconitemDoubleClicked(self, item=None):self.currIndex = self.mImgList.index(ustr(os.path.join(item.toolTip())))filename = self.mImgList[self.currIndex]if filename:self.mImgList5 = self.indexTo5Files(self.currIndex)# self.additems5(None)self.loadFile(filename)def CanvasSizeChange(self):if len(self.mImgList) > 0 and self.imageSlider.hasFocus():self.zoomWidget.setValue(self.imageSlider.value())def shapeSelectionChanged(self, selected_shapes):self._noSelectionSlot = Truefor shape in self.canvas.selectedShapes:shape.selected = Falseself.labelList.clearSelection()self.indexList.clearSelection()self.canvas.selectedShapes = selected_shapesfor shape in self.canvas.selectedShapes:shape.selected = Trueself.shapesToItems[shape].setSelected(True)self.shapesToItemsbox[shape].setSelected(True)index = self.labelList.indexFromItem(self.shapesToItems[shape]).row()self.indexList.item(index).setSelected(True)self.labelList.scrollToItem(self.currentItem()) # QAbstractItemView.EnsureVisible# map current label item to index item and select itindex = self.labelList.indexFromItem(self.currentItem()).row()self.indexList.scrollToItem(self.indexList.item(index))self.BoxList.scrollToItem(self.currentBox())if self.kie_mode:if len(self.canvas.selectedShapes) == 1 and self.keyList.count() > 0:selected_key_item_row = self.keyList.findItemsByLabel(self.canvas.selectedShapes[0].key_cls, get_row=True)if (isinstance(selected_key_item_row, list)and len(selected_key_item_row) == 0):key_text = self.canvas.selectedShapes[0].key_clsitem = self.keyList.createItemFromLabel(key_text)self.keyList.addItem(item)rgb = self._get_rgb_by_label(key_text, self.kie_mode)self.keyList.setItemLabel(item, key_text, rgb)selected_key_item_row = self.keyList.findItemsByLabel(self.canvas.selectedShapes[0].key_cls, get_row=True)self.keyList.setCurrentRow(selected_key_item_row)self._noSelectionSlot = Falsen_selected = len(selected_shapes)self.actions.singleRere.setEnabled(n_selected)self.actions.cellreRec.setEnabled(n_selected)self.actions.delete.setEnabled(n_selected)self.actions.copy.setEnabled(n_selected)self.actions.edit.setEnabled(n_selected == 1)self.actions.lock.setEnabled(n_selected)self.actions.change_cls.setEnabled(n_selected)self.actions.expand.setEnabled(n_selected)def addLabel(self, shape):shape.paintLabel = self.displayLabelOption.isChecked()shape.paintIdx = self.displayIndexOption.isChecked()item = HashableQListWidgetItem(shape.label)# current difficult checkbox is disenble# item.setFlags(item.flags() | Qt.ItemIsUserCheckable)# item.setCheckState(Qt.Unchecked) if shape.difficult else item.setCheckState(Qt.Checked)# Checked means difficult is False# item.setBackground(generateColorByText(shape.label))self.itemsToShapes[item] = shapeself.shapesToItems[shape] = item# add current label item index before label stringcurrent_index = QListWidgetItem(str(self.labelList.count()))current_index.setTextAlignment(Qt.AlignHCenter)self.indexList.addItem(current_index)self.labelList.addItem(item)# print('item in add label is ',[(p.x(), p.y()) for p in shape.points], shape.label)# ADD for boxitem = HashableQListWidgetItem(str([(int(p.x()), int(p.y())) for p in shape.points]))self.itemsToShapesbox[item] = shapeself.shapesToItemsbox[shape] = itemself.BoxList.addItem(item)for action in self.actions.onShapesPresent:action.setEnabled(True)self.updateComboBox()# update show countingself.BoxListDock.setWindowTitle(self.BoxListDockName + f" ({self.BoxList.count()})")self.labelListDock.setWindowTitle(self.labelListDockName + f" ({self.labelList.count()})")def remLabels(self, shapes):if shapes is None:# print('rm empty label')returnfor shape in shapes:item = self.shapesToItems[shape]self.labelList.takeItem(self.labelList.row(item))del self.shapesToItems[shape]del self.itemsToShapes[item]self.updateComboBox()# ADD:item = self.shapesToItemsbox[shape]self.BoxList.takeItem(self.BoxList.row(item))del self.shapesToItemsbox[shape]del self.itemsToShapesbox[item]self.updateComboBox()self.updateIndexList()def loadLabels(self, shapes):s = []shape_index = 0for label, points, line_color, key_cls, difficult in shapes:shape = Shape(label=label,line_color=line_color,key_cls=key_cls,font_family=self.label_font_family,)for x, y in points:# Ensure the labels are within the bounds of the image. If not, fix them.x, y, snapped = self.canvas.snapPointToCanvas(x, y)if snapped:self.setDirty()shape.addPoint(QPointF(x, y))shape.difficult = difficultshape.idx = shape_indexshape_index += 1# shape.locked = Falseshape.close()s.append(shape)self._update_shape_color(shape)self.addLabel(shape)self.updateComboBox()self.canvas.loadShapes(s)def singleLabel(self, shape):if shape is None:# print('rm empty label')returnitem = self.shapesToItems[shape]item.setText(shape.label)self.updateComboBox()# ADD:item = self.shapesToItemsbox[shape]item.setText(str([(int(p.x()), int(p.y())) for p in shape.points]))self.updateComboBox()def updateComboBox(self):# Get the unique labels and add them to the Combobox.itemsTextList = [str(self.labelList.item(i).text()) for i in range(self.labelList.count())]uniqueTextList = list(set(itemsTextList))# Add a null row for showing all the labelsuniqueTextList.append("")uniqueTextList.sort()# self.comboBox.update_items(uniqueTextList)def updateIndexList(self):self.indexList.clear()for i in range(self.labelList.count()):string = QListWidgetItem(str(i))string.setTextAlignment(Qt.AlignHCenter)self.indexList.addItem(string)def saveLabels(self, annotationFilePath, mode="Auto"):# Mode is Auto means that labels will be loaded from self.result_dic totally, which is the output of ocr modelannotationFilePath = ustr(annotationFilePath)def format_shape(s):# print('s in saveLabels is ',s)return dict(label=s.label, # strline_color=s.line_color.getRgb(),fill_color=s.fill_color.getRgb(),points=[(int(p.x()), int(p.y())) for p in s.points], # QPonitFdifficult=s.difficult,key_cls=s.key_cls,) # boolif mode == "Auto":shapes = []else:shapes = [format_shape(shape)for shape in self.canvas.shapesif shape.line_color != DEFAULT_LOCK_COLOR]# Can add differrent annotation formats herefor box in self.result_dic:trans_dic = {"label": box[1][0], "points": box[0], "difficult": False}if self.kie_mode:if len(box) == 3:trans_dic.update({"key_cls": box[2]})else:trans_dic.update({"key_cls": "None"})if trans_dic["label"] == "" and mode == "Auto":continueshapes.append(trans_dic)try:trans_dic = []for box in shapes:trans_dict = {"transcription": box["label"],"points": box["points"],"difficult": box["difficult"],}if self.kie_mode:trans_dict.update({"key_cls": box["key_cls"]})trans_dic.append(trans_dict)self.PPlabel[annotationFilePath] = trans_dicif mode == "Auto":self.Cachelabel[annotationFilePath] = trans_dic# else:# self.labelFile.save(annotationFilePath, shapes, self.filePath, self.imageData,# self.lineColor.getRgb(), self.fillColor.getRgb())# print('Image:{0} -> Annotation:{1}'.format(self.filePath, annotationFilePath))return Trueexcept Exception:self.errorMessage("Error saving label data", "Error saving label data")return Falsedef copySelectedShape(self):for shape in self.canvas.copySelectedShape():self.addLabel(shape)# fix copy and delete# self.shapeSelectionChanged(True)def move_scrollbar(self, value):self.labelListBar.setValue(int(value))self.indexListBar.setValue(int(value))def labelSelectionChanged(self):if self._noSelectionSlot:returnif self.canvas.editing():selected_shapes = []for item in self.labelList.selectedItems():selected_shapes.append(self.itemsToShapes[item])if selected_shapes:self.canvas.selectShapes(selected_shapes)else:self.canvas.deSelectShape()def indexSelectionChanged(self):if self._noSelectionSlot:returnif self.canvas.editing():selected_shapes = []for item in self.indexList.selectedItems():# map index item to label itemindex = self.indexList.indexFromItem(item).row()item = self.labelList.item(index)selected_shapes.append(self.itemsToShapes[item])if selected_shapes:self.canvas.selectShapes(selected_shapes)else:self.canvas.deSelectShape()def boxSelectionChanged(self):if self._noSelectionSlot:# self.BoxList.scrollToItem(self.currentBox(), QAbstractItemView.PositionAtCenter)returnif self.canvas.editing():selected_shapes = []for item in self.BoxList.selectedItems():selected_shapes.append(self.itemsToShapesbox[item])if selected_shapes:self.canvas.selectShapes(selected_shapes)else:self.canvas.deSelectShape()def labelItemChanged(self, item):# avoid accidentally triggering the itemChanged siganl with unhashable item# Unknown trigger conditionif isinstance(item, HashableQListWidgetItem):shape = self.itemsToShapes[item]label = item.text()if label != shape.label:shape.label = item.text()# shape.line_color = generateColorByText(shape.label)self.setDirty()elif not ((item.checkState() == Qt.Unchecked) ^ (not shape.difficult)):shape.difficult = True if item.checkState() == Qt.Unchecked else Falseself.setDirty()else: # User probably changed item visibilityself.canvas.setShapeVisible(shape, True) # item.checkState() == Qt.Checked# self.actions.save.setEnabled(True)else:print("enter labelItemChanged slot with unhashable item: ", item, item.text())def drag_drop_happened(self):"""label list drag drop signal slot"""# print('___________________drag_drop_happened_______________')# should only select single itemfor item in self.labelList.selectedItems():newIndex = self.labelList.indexFromItem(item).row()# only support drag_drop one itemassert len(self.canvas.selectedShapes) > 0for shape in self.canvas.selectedShapes:selectedShapeIndex = shape.idxif newIndex == selectedShapeIndex:return# move corresponding item in shape listshape = self.canvas.shapes.pop(selectedShapeIndex)self.canvas.shapes.insert(newIndex, shape)# update bbox indexself.canvas.updateShapeIndex()# boxList update simultaneouslyitem = self.BoxList.takeItem(selectedShapeIndex)self.BoxList.insertItem(newIndex, item)# changes happenself.setDirty()# Callback functions:def newShape(self, value=True):"""Pop-up and give focus to the label editor.position MUST be in global coordinates."""if len(self.labelHist) > 0:self.labelDialog = LabelDialog(parent=self, listItem=self.labelHist)if value:text = self.labelDialog.popUp(text=self.prevLabelText)self.lastLabel = textelse:text = self.prevLabelTextif text is not None:self.prevLabelText = self.stringBundle.getString("tempLabel")shape = self.canvas.setLastLabel(text, None, None, None) # generate_color, generate_colorif self.kie_mode:key_text, _ = self.keyDialog.popUp(self.key_previous_text)if key_text is not None:shape = self.canvas.setLastLabel(text, None, None, key_text) # generate_color, generate_colorself.key_previous_text = key_textif not self.keyList.findItemsByLabel(key_text):item = self.keyList.createItemFromLabel(key_text)self.keyList.addItem(item)rgb = self._get_rgb_by_label(key_text, self.kie_mode)self.keyList.setItemLabel(item, key_text, rgb)self._update_shape_color(shape)self.keyDialog.addLabelHistory(key_text)self.addLabel(shape)if self.beginner(): # Switch to edit mode.self.canvas.setEditing(True)self.actions.create.setEnabled(True)self.actions.createpoly.setEnabled(True)self.actions.undoLastPoint.setEnabled(False)self.actions.undo.setEnabled(True)else:self.actions.editMode.setEnabled(True)self.setDirty()if self.autoReRecognitionOption.isChecked():self.reRecognition()else:# self.canvas.undoLastLine()self.canvas.resetAllLines()def _update_shape_color(self, shape):r, g, b = self._get_rgb_by_label(shape.key_cls, self.kie_mode)shape.line_color = QColor(r, g, b)shape.vertex_fill_color = QColor(r, g, b)shape.hvertex_fill_color = QColor(255, 255, 255)shape.fill_color = QColor(r, g, b, 32)shape.select_line_color = QColor(self.selected_shape_color[0],self.selected_shape_color[1],self.selected_shape_color[2],)shape.select_fill_color = QColor(r, g, b, 32)def _get_rgb_by_label(self, label, kie_mode):shift_auto_shape_color = 2 # use for random colorif kie_mode and label != "None":item = self.keyList.findItemsByLabel(label)[0]label_id = self.keyList.indexFromItem(item).row() + 1label_id += shift_auto_shape_colorreturn LABEL_COLORMAP[label_id % len(LABEL_COLORMAP)]else:return (0, 255, 0)def scrollRequest(self, delta, orientation):units = -delta / (8 * 15)bar = self.scrollBars[orientation]bar.setValue(int(bar.value() + bar.singleStep() * units))def setZoom(self, value):self.actions.fitWidth.setChecked(False)self.actions.fitWindow.setChecked(False)self.zoomMode = self.MANUAL_ZOOMself.zoomWidget.setValue(int(value))def addZoom(self, increment=10):self.setZoom(int(self.zoomWidget.value() + increment))self.imageSlider.setValue(int(self.zoomWidget.value() + increment)) # set zoom slider valuedef zoomRequest(self, delta, pos: QPoint = None):# get the current scrollbar positions# calculate the percentages ~ coordinatesh_bar = self.scrollBars[Qt.Horizontal]v_bar = self.scrollBars[Qt.Vertical]# get the current maximum, to know the difference after zoomingh_bar_max = h_bar.maximum()v_bar_max = v_bar.maximum()# get the cursor position and canvas size# calculate the desired movement from 0 to 1# where 0 = move left# 1 = move right# up and down analogousif pos is None:cursor = QCursor()pos = cursor.pos()relative_pos = QWidget.mapFromGlobal(self, pos)cursor_x = relative_pos.x()cursor_y = relative_pos.y()w = self.scrollArea.width()h = self.scrollArea.height()# the scaling from 0 to 1 has some padding# you don't have to hit the very leftmost pixel for a maximum-left movementmargin = 0.1move_x = (cursor_x - margin * w) / (w - 2 * margin * w)move_y = (cursor_y - margin * h) / (h - 2 * margin * h)# clamp the values from 0 to 1move_x = min(max(move_x, 0), 1)move_y = min(max(move_y, 0), 1)# zoom inunits = delta / (8 * 15)scale = 10self.addZoom(scale * units)# get the difference in scrollbar values# this is how far we can moved_h_bar_max = h_bar.maximum() - h_bar_maxd_v_bar_max = v_bar.maximum() - v_bar_max# get the new scrollbar valuesnew_h_bar_value = h_bar.value() + move_x * d_h_bar_maxnew_v_bar_value = v_bar.value() + move_y * d_v_bar_maxh_bar.setValue(int(new_h_bar_value))v_bar.setValue(int(new_v_bar_value))def setFitWindow(self, value=True):if value:self.actions.fitWidth.setChecked(False)self.zoomMode = self.FIT_WINDOW if value else self.MANUAL_ZOOMself.adjustScale()def setFitWidth(self, value=True):if value:self.actions.fitWindow.setChecked(False)self.zoomMode = self.FIT_WIDTH if value else self.MANUAL_ZOOMself.adjustScale()def togglePolygons(self, value):for item, shape in self.itemsToShapes.items():self.canvas.setShapeVisible(shape, value)def loadFile(self, filePath=None, isAdjustScale=True):"""Load the specified file, or the last opened file if None."""self.canvas.shape_move_index = Noneif self.dirty:self.mayContinue()self.resetState()self.canvas.setEnabled(False)if filePath is None:filePath = self.settings.get(SETTING_FILENAME)# Make sure that filePath is a regular python string, rather than QStringfilePath = ustr(filePath)# Fix bug: An index error after select a directory when open a new file.unicodeFilePath = ustr(filePath)# unicodeFilePath = os.path.abspath(unicodeFilePath)# Tzutalin 20160906 : Add file list and dock to move faster# Highlight the file itemif unicodeFilePath and self.fileListWidget.count() > 0:if unicodeFilePath in self.mImgList:index = self.mImgList.index(unicodeFilePath)fileWidgetItem = self.fileListWidget.item(index)print("unicodeFilePath is", unicodeFilePath)fileWidgetItem.setSelected(True)self.iconlist.clear()self.additems5(None)for i in range(5):item_tooltip = self.iconlist.item(i).toolTip()# print(i,"---",item_tooltip)if item_tooltip == ustr(filePath):titem = self.iconlist.item(i)titem.setSelected(True)self.iconlist.scrollToItem(titem)breakelse:self.fileListWidget.clear()self.mImgList.clear()self.iconlist.clear()# if unicodeFilePath and self.iconList.count() > 0:# if unicodeFilePath in self.mImgList:if unicodeFilePath and os.path.exists(unicodeFilePath):self.canvas.verified = Falsecvimg = cv2.imdecode(np.fromfile(unicodeFilePath, dtype=np.uint8), 1)height, width, depth = cvimg.shapecvimg = cv2.cvtColor(cvimg, cv2.COLOR_BGR2RGB)image = QImage(cvimg.data, width, height, width * depth, QImage.Format_RGB888)if image.isNull():self.errorMessage("Error opening file","<p>Make sure <i>%s</i> is a valid image file." % unicodeFilePath,)self.status("Error reading %s" % unicodeFilePath)return Falseself.status("Loaded %s" % os.path.basename(unicodeFilePath))self.image = imageself.filePath = unicodeFilePathself.canvas.loadPixmap(QPixmap.fromImage(image))if self.validFilestate(filePath) is True:self.setClean()else:self.dirty = Falseself.actions.save.setEnabled(True)if len(self.canvas.lockedShapes) != 0:self.actions.save.setEnabled(True)self.setDirty()self.canvas.setEnabled(True)if isAdjustScale:self.adjustScale(initial=True)self.paintCanvas()self.addRecentFile(self.filePath)self.toggleActions(True)self.showBoundingBoxFromPPlabel(filePath)self.setWindowTitle(__appname__ + " " + filePath)# Default : select last item if there is at least one itemif self.labelList.count():self.labelList.setCurrentItem(self.labelList.item(self.labelList.count() - 1))self.labelList.item(self.labelList.count() - 1).setSelected(True)self.indexList.item(self.labelList.count() - 1).setSelected(True)# show file list image countselect_indexes = self.fileListWidget.selectedIndexes()if len(select_indexes) > 0:self.fileDock.setWindowTitle(self.fileListName + f" ({select_indexes[0].row() + 1}"f"/{self.fileListWidget.count()})")# update show countingself.BoxListDock.setWindowTitle(self.BoxListDockName + f" ({self.BoxList.count()})")self.labelListDock.setWindowTitle(self.labelListDockName + f" ({self.labelList.count()})")self.canvas.setFocus(True)if self.bbox_auto_zoom_center:if len(self.canvas.shapes) > 0:(center_x,center_y,shape_area,) = polygon_bounding_box_center_and_area(self.canvas.shapes[0].points)if shape_area < 30000:zoom_value = 120 * map_value(shape_area, 100, 30000, 20, 0)self.zoomRequest(zoom_value, QPoint(center_x, center_y))# print(" =========> ", shape_area, " ==> ", zoom_value)return Truereturn Falsedef showBoundingBoxFromPPlabel(self, filePath):width, height = self.image.width(), self.image.height()imgidx = self.getImglabelidx(filePath)shapes = []# box['ratio'] of the shapes saved in lockedShapes contains the ratio of the# four corner coordinates of the shapes to the height and width of the imagefor box in self.canvas.lockedShapes:key_cls = "None" if not self.kie_mode else box["key_cls"]if self.canvas.isInTheSameImage:shapes.append((box["transcription"],[[s[0] * width, s[1] * height] for s in box["ratio"]],DEFAULT_LOCK_COLOR,key_cls,box["difficult"],))else:shapes.append(("锁定框：待检测",[[s[0] * width, s[1] * height] for s in box["ratio"]],DEFAULT_LOCK_COLOR,key_cls,box["difficult"],))if imgidx in self.PPlabel.keys():for box in self.PPlabel[imgidx]:key_cls = "None" if not self.kie_mode else box.get("key_cls", "None")shapes.append((box["transcription"],box["points"],None,key_cls,box.get("difficult", False),))if shapes != []:self.loadLabels(shapes)self.canvas.verified = Falsedef validFilestate(self, filePath):if filePath in self.fileStatedict.keys() and self.fileStatedict[filePath] == 1:return Trueelif (self.getImglabelidx(filePath) in self.fileStatedict.keys()and self.fileStatedict[self.getImglabelidx(filePath)] == 1):return Trueelse:return Falsedef resizeEvent(self, event):if (self.canvasand not self.image.isNull()and self.zoomMode != self.MANUAL_ZOOM):self.adjustScale()super(MainWindow, self).resizeEvent(event)def paintCanvas(self):assert not self.image.isNull(), "cannot paint null image"self.canvas.scale = 0.01 * self.zoomWidget.value()self.canvas.adjustSize()self.canvas.update()def adjustScale(self, initial=False):value = self.scalers[self.FIT_WINDOW if initial else self.zoomMode]()self.zoomWidget.setValue(int(100 * value))self.imageSlider.setValue(self.zoomWidget.value()) # set zoom slider valuedef scaleFitWindow(self):"""Figure out the size of the pixmap in order to fit the main widget."""e = 2.0 # So that no scrollbars are generated.w1 = self.centralWidget().width() - eh1 = self.centralWidget().height() - e - 110a1 = w1 / h1# Calculate a new scale value based on the pixmap's aspect ratio.w2 = self.canvas.pixmap.width() - 0.0h2 = self.canvas.pixmap.height() - 0.0a2 = w2 / h2return w1 / w2 if a2 >= a1 else h1 / h2def scaleFitWidth(self):# The epsilon does not seem to work too well here.w = self.centralWidget().width() - 2.0return w / self.canvas.pixmap.width()def closeEvent(self, event):if not self.mayContinue():event.ignore()else:settings = self.settings# If it loads images from dir, don't load it at the beginningif self.dirname is None:settings[SETTING_FILENAME] = self.filePath if self.filePath else ""else:settings[SETTING_FILENAME] = ""settings[SETTING_WIN_SIZE] = self.size()settings[SETTING_WIN_POSE] = self.pos()settings[SETTING_WIN_STATE] = self.saveState()settings[SETTING_LINE_COLOR] = self.lineColorsettings[SETTING_FILL_COLOR] = self.fillColorsettings[SETTING_RECENT_FILES] = self.recentFilessettings[SETTING_ADVANCE_MODE] = not self._beginnerif self.defaultSaveDir and os.path.exists(self.defaultSaveDir):settings[SETTING_SAVE_DIR] = ustr(self.defaultSaveDir)else:settings[SETTING_SAVE_DIR] = ""if self.lastOpenDir and os.path.exists(self.lastOpenDir):settings[SETTING_LAST_OPEN_DIR] = self.lastOpenDirelse:settings[SETTING_LAST_OPEN_DIR] = ""settings[SETTING_PAINT_LABEL] = self.displayLabelOption.isChecked()settings[SETTING_PAINT_INDEX] = self.displayIndexOption.isChecked()settings[SETTING_DRAW_SQUARE] = self.drawSquaresOption.isChecked()settings.save()try:self.saveLabelFile()except Exception:passdef loadRecent(self, filename):if self.mayContinue():print(filename, "======")self.loadFile(filename)def scanAllImages(self, folderPath):extensions = [".%s" % fmt.data().decode("ascii").lower()for fmt in QImageReader.supportedImageFormats()]images = []for file in os.listdir(folderPath):if file.lower().endswith(tuple(extensions)):relativePath = os.path.join(folderPath, file)path = ustr(os.path.abspath(relativePath))images.append(path)if self.img_list_natural_sort:natural_sort(images, key=lambda x: x.lower())else:images.sort()return imagesdef openDirDialog(self, _value=False, dirpath=None, silent=False):if not self.mayContinue():returndefaultOpenDirPath = dirpath if dirpath else "."if self.lastOpenDir and os.path.exists(self.lastOpenDir):defaultOpenDirPath = self.lastOpenDirelse:defaultOpenDirPath = (os.path.dirname(self.filePath) if self.filePath else ".")if not silent:targetDirPath = ustr(QFileDialog.getExistingDirectory(self,"%s - Open Directory" % __appname__,defaultOpenDirPath,QFileDialog.ShowDirsOnly | QFileDialog.DontResolveSymlinks,))else:targetDirPath = ustr(defaultOpenDirPath)self.lastOpenDir = targetDirPathself.importDirImages(targetDirPath)def openDatasetDirDialog(self):if self.lastOpenDir and os.path.exists(self.lastOpenDir):if platform.system() == "Windows":os.startfile(self.lastOpenDir)else:os.system("open " + os.path.normpath(self.lastOpenDir))defaultOpenDirPath = self.lastOpenDirelse:if self.lang == "ch":self.msgBox.warning(self, "提示", "\n 原文件夹已不存在,请从新选择数据集路径!")else:self.msgBox.warning(self,"Warn","\n The original folder no longer exists, please choose the data set path again!",)self.actions.open_dataset_dir.setEnabled(False)defaultOpenDirPath = (os.path.dirname(self.filePath) if self.filePath else ".")def init_key_list(self, label_dict):if not self.kie_mode:return# load key_clsfor image, info in label_dict.items():for box in info:if "key_cls" not in box:box.update({"key_cls": "None"})self.existed_key_cls_set.add(box["key_cls"])if len(self.existed_key_cls_set) > 0:for key_text in self.existed_key_cls_set:if not self.keyList.findItemsByLabel(key_text):item = self.keyList.createItemFromLabel(key_text)self.keyList.addItem(item)rgb = self._get_rgb_by_label(key_text, self.kie_mode)self.keyList.setItemLabel(item, key_text, rgb)if self.keyDialog is None:# key list dialogself.keyDialog = KeyDialog(text=self.key_dialog_tip,parent=self,labels=self.existed_key_cls_set,sort_labels=True,show_text_field=True,completion="startswith",fit_to_content={"column": True, "row": False},flags=None,)def importDirImages(self, dirpath, isDelete=False):if not self.mayContinue() or not dirpath:returnif self.defaultSaveDir and self.defaultSaveDir != dirpath:self.saveLabelFile()if not isDelete:self.loadFilestate(dirpath)self.PPlabelpath = dirpath + "/Label.txt"self.PPlabel = self.loadLabelFile(self.PPlabelpath)self.Cachelabelpath = dirpath + "/Cache.cach"self.Cachelabel = self.loadLabelFile(self.Cachelabelpath)if self.Cachelabel:self.PPlabel = dict(self.Cachelabel, **self.PPlabel)self.init_key_list(self.PPlabel)self.lastOpenDir = dirpathself.dirname = dirpathself.defaultSaveDir = dirpathself.statusBar().showMessage("%s started. Annotation will be saved to %s"% (__appname__, self.defaultSaveDir))self.statusBar().show()imgListCurrIndex = Noneif self.filePath != None:imgListCurrIndex = self.mImgList.index(self.filePath)self.filePath = Noneself.fileListWidget.clear()self.mImgList = self.scanAllImages(dirpath)self.mImgList5 = self.mImgList[:5]self.openNextImg(imgListCurrIndex=imgListCurrIndex)doneicon = newIcon("done")closeicon = newIcon("close")for imgPath in self.mImgList:filename = os.path.basename(imgPath)if self.validFilestate(imgPath) is True:item = QListWidgetItem(doneicon, filename)else:item = QListWidgetItem(closeicon, filename)self.fileListWidget.addItem(item)print("DirPath in importDirImages is", dirpath)self.iconlist.clear()self.additems5(dirpath)self.changeFileFolder = Trueself.haveAutoReced = Falseself.AutoRecognition.setEnabled(True)self.reRecogButton.setEnabled(True)self.tableRecButton.setEnabled(True)self.actions.AutoRec.setEnabled(True)self.actions.reRec.setEnabled(True)self.actions.tableRec.setEnabled(True)self.actions.open_dataset_dir.setEnabled(True)self.actions.rotateLeft.setEnabled(True)self.actions.rotateRight.setEnabled(True)fileListWidgetCurrentRow = 0if imgListCurrIndex is not None:fileListWidgetCurrentRow = imgListCurrIndexif fileListWidgetCurrentRow >= self.fileListWidget.count():fileListWidgetCurrentRow = fileListWidgetCurrentRow - 1self.fileListWidget.setCurrentRow(fileListWidgetCurrentRow) # set list index to firstself.fileDock.setWindowTitle(self.fileListName + f" (1/{self.fileListWidget.count()})") # show image countdef openPrevImg(self, _value=False):if len(self.mImgList) <= 0:returnif self.filePath is None:returncurrIndex = self.mImgList.index(self.filePath)self.mImgList5 = self.mImgList[:5]if currIndex - 1 >= 0:filename = self.mImgList[currIndex - 1]self.mImgList5 = self.indexTo5Files(currIndex - 1)if filename:self.loadFile(filename)def openNextImg(self, _value=False, imgListCurrIndex=None):if not self.mayContinue():returnif len(self.mImgList) <= 0:returnfilename = Noneif self.filePath is None and imgListCurrIndex is None:filename = self.mImgList[0]self.mImgList5 = self.mImgList[:5]else:if imgListCurrIndex is None:currIndex = self.mImgList.index(self.filePath)else:currIndex = imgListCurrIndex - 1if currIndex + 1 < len(self.mImgList):filename = self.mImgList[currIndex + 1]self.mImgList5 = self.indexTo5Files(currIndex + 1)else:filename = self.mImgList[currIndex]self.mImgList5 = self.indexTo5Files(currIndex)if filename:print("file name in openNext is ", filename)self.loadFile(filename)def updateFileListIcon(self, filename):passdef saveFile(self, _value=False, mode="Manual"):# Manual mode is used for users click "Save" manually,which will change the state of the imageif self.filePath:imgidx = self.getImglabelidx(self.filePath)self._saveFile(imgidx, mode=mode)def saveLockedShapes(self):self.canvas.lockedShapes = []self.canvas.selectedShapes = []for s in self.canvas.shapes:if s.line_color == DEFAULT_LOCK_COLOR:self.canvas.selectedShapes.append(s)self.lockSelectedShape()for s in self.canvas.shapes:if s.line_color == DEFAULT_LOCK_COLOR:self.canvas.selectedShapes.remove(s)self.canvas.shapes.remove(s)def _saveFile(self, annotationFilePath, mode="Manual"):if len(self.canvas.lockedShapes) != 0:self.saveLockedShapes()if mode == "Manual":self.result_dic_locked = []img = cv2.imdecode(np.fromfile(self.filePath, dtype=np.uint8), cv2.IMREAD_COLOR)width, height = self.image.width(), self.image.height()for shape in self.canvas.lockedShapes:box = [[int(p[0] * width), int(p[1] * height)] for p in shape["ratio"]]# assert len(box) == 4result = [(shape["transcription"], 1)]result.insert(0, box)self.result_dic_locked.append(result)self.result_dic += self.result_dic_lockedself.result_dic_locked = []if annotationFilePath and self.saveLabels(annotationFilePath, mode=mode):self.setClean()self.statusBar().showMessage("Saved to %s" % annotationFilePath)self.statusBar().show()currIndex = self.mImgList.index(self.filePath)item = self.fileListWidget.item(currIndex)item.setIcon(newIcon("done"))self.fileStatedict[self.getImglabelidx(self.filePath)] = 1if len(self.fileStatedict) % self.autoSaveNum == 0:self.saveFilestate()self.savePPlabel(mode="Auto")self.fileListWidget.insertItem(int(currIndex), item)if not self.canvas.isInTheSameImage:self.openNextImg()self.actions.saveRec.setEnabled(True)self.actions.saveLabel.setEnabled(True)self.actions.exportJSON.setEnabled(True)elif mode == "Auto":if annotationFilePath and self.saveLabels(annotationFilePath, mode=mode):self.setClean()self.statusBar().showMessage("Saved to %s" % annotationFilePath)self.statusBar().show()def closeFile(self, _value=False):if not self.mayContinue():returnself.resetState()self.setClean()self.toggleActions(False)self.canvas.setEnabled(False)self.actions.saveAs.setEnabled(False)def deleteImg(self):deletePath = self.filePathif deletePath is not None:deleteInfo = self.deleteImgDialog()if deleteInfo == QMessageBox.Yes:if platform.system() == "Windows":# from win32com import shell, shellcon# shell.SHFileOperation((0, shellcon.FO_DELETE, deletePath, None,# shellcon.FOF_SILENT | shellcon.FOF_ALLOWUNDO | shellcon.FOF_NOCONFIRMATION,# None, None))os.remove(deletePath)# linuxelif platform.system() == "Linux":cmd = "trash " + deletePathos.system(cmd)# macOSelif platform.system() == "Darwin":import subprocessabsPath = (os.path.abspath(deletePath).replace("\\", "\\\\").replace('"', '\\"'))cmd = ["osascript","-e",'tell app "Finder" to move {the POSIX file "'+ absPath+ '"} to trash',]print(cmd)subprocess.call(cmd, stdout=open(os.devnull, "w"))if self.filePath in self.fileStatedict.keys():self.fileStatedict.pop(self.filePath)imgidx = self.getImglabelidx(self.filePath)if imgidx in self.PPlabel.keys():self.PPlabel.pop(imgidx)self.importDirImages(self.lastOpenDir, isDelete=True)def deleteImgDialog(self):yes, cancel = QMessageBox.Yes, QMessageBox.Cancelmsg = "The image will be deleted to the recycle bin"return QMessageBox.warning(self, "Attention", msg, yes | cancel)def resetAll(self):self.settings.reset()self.close()proc = QProcess()proc.startDetached(os.path.abspath(__file__))def mayContinue(self): #if not self.dirty:return Trueelse:if self.autoSaveUnsavedChangesOption.isChecked():self.canvas.isInTheSameImage = Trueself.saveFile()self.canvas.isInTheSameImage = Falsereturn TruediscardChanges = self.discardChangesDialog()if discardChanges == QMessageBox.No:return Trueelif discardChanges == QMessageBox.Yes:self.canvas.isInTheSameImage = Trueself.saveFile()self.canvas.isInTheSameImage = Falsereturn Trueelse:return Falsedef discardChangesDialog(self):yes, no, cancel = QMessageBox.Yes, QMessageBox.No, QMessageBox.Cancelif self.lang == "ch":msg = '您有未保存的变更, 您想保存再继续吗?\n点击 "No" 丢弃所有未保存的变更.'else:msg = 'You have unsaved changes, would you like to save them and proceed?\nClick "No" to undo all changes.'return QMessageBox.warning(self, "Attention", msg, yes | no | cancel)def errorMessage(self, title, message):return QMessageBox.critical(self, title, "<p><b>%s</b></p>%s" % (title, message))def currentPath(self):return os.path.dirname(self.filePath) if self.filePath else "."def chooseColor(self):color = self.colorDialog.getColor(self.lineColor, "Choose line color", default=DEFAULT_LINE_COLOR)if color:self.lineColor = colorShape.line_color = colorself.canvas.setDrawingColor(color)self.canvas.update()self.setDirty()def deleteSelectedShape(self):self.remLabels(self.canvas.deleteSelected())self.actions.undo.setEnabled(True)self.setDirty()if self.noShapes():for action in self.actions.onShapesPresent:action.setEnabled(False)self.BoxListDock.setWindowTitle(self.BoxListDockName + f" ({self.BoxList.count()})")self.labelListDock.setWindowTitle(self.labelListDockName + f" ({self.labelList.count()})")def chshapeLineColor(self):color = self.colorDialog.getColor(self.lineColor, "Choose line color", default=DEFAULT_LINE_COLOR)if color:for shape in self.canvas.selectedShapes:shape.line_color = colorself.canvas.update()self.setDirty()def chshapeFillColor(self):color = self.colorDialog.getColor(self.fillColor, "Choose fill color", default=DEFAULT_FILL_COLOR)if color:for shape in self.canvas.selectedShapes:shape.fill_color = colorself.canvas.update()self.setDirty()def copyShape(self):self.canvas.endMove(copy=True)self.addLabel(self.canvas.selectedShape)self.setDirty()def moveShape(self):self.canvas.endMove(copy=False)self.setDirty()def loadPredefinedClasses(self, predefClassesFile):if os.path.exists(predefClassesFile) is True:with codecs.open(predefClassesFile, "r", "utf8") as f:for line in f:line = line.strip()if self.labelHist is None:self.labelHist = [line]else:self.labelHist.append(line)def togglePaintLabelsOption(self):self.displayIndexOption.setChecked(False)for shape in self.canvas.shapes:shape.paintLabel = self.displayLabelOption.isChecked()shape.paintIdx = self.displayIndexOption.isChecked()self.canvas.repaint()def togglePaintIndexOption(self):self.displayLabelOption.setChecked(False)for shape in self.canvas.shapes:shape.paintLabel = self.displayLabelOption.isChecked()shape.paintIdx = self.displayIndexOption.isChecked()self.canvas.repaint()def toogleDrawSquare(self):self.canvas.setDrawingShapeToSquare(self.drawSquaresOption.isChecked())def additems(self, dirpath):for file in self.mImgList:pix = QPixmap(file)_, filename = os.path.split(file)filename, _ = os.path.splitext(filename)item = QListWidgetItem(QIcon(pix.scaled(100, 100, Qt.IgnoreAspectRatio, Qt.FastTransformation)),filename[:10],)item.setToolTip(file)self.iconlist.addItem(item)def additems5(self, dirpath):for file in self.mImgList5:pix = QPixmap(file)_, filename = os.path.split(file)filename, _ = os.path.splitext(filename)pfilename = filename[:10]if len(pfilename) < 10:lentoken = 12 - len(pfilename)prelen = lentoken // 2bfilename = prelen * " " + pfilename + (lentoken - prelen) * " "# item = QListWidgetItem(QIcon(pix.scaled(100, 100, Qt.KeepAspectRatio, Qt.SmoothTransformation)),filename[:10])item = QListWidgetItem(QIcon(pix.scaled(100, 100, Qt.IgnoreAspectRatio, Qt.FastTransformation)),pfilename,)# item.setForeground(QBrush(Qt.white))item.setToolTip(file)self.iconlist.addItem(item)owidth = 0for index in range(len(self.mImgList5)):item = self.iconlist.item(index)itemwidget = self.iconlist.visualItemRect(item)owidth += itemwidget.width()self.iconlist.setMinimumWidth(owidth + 50)def gen_quad_from_poly(self, poly):"""Generate min area quad from poly."""point_num = poly.shape[0]min_area_quad = np.zeros((4, 2), dtype=np.float32)rect = cv2.minAreaRect(poly.astype(np.int32)) # (center (x,y), (width, height), angle of rotation)box = np.array(cv2.boxPoints(rect))first_point_idx = 0min_dist = 1e4for i in range(4):dist = (np.linalg.norm(box[(i + 0) % 4] - poly[0])+ np.linalg.norm(box[(i + 1) % 4] - poly[point_num // 2 - 1])+ np.linalg.norm(box[(i + 2) % 4] - poly[point_num // 2])+ np.linalg.norm(box[(i + 3) % 4] - poly[-1]))if dist < min_dist:min_dist = distfirst_point_idx = ifor i in range(4):min_area_quad[i] = box[(first_point_idx + i) % 4]bbox_new = min_area_quad.tolist()bbox = []for box in bbox_new:box = list(map(int, box))bbox.append(box)return bboxdef getImglabelidx(self, filePath):if platform.system() == "Windows":spliter = "\\"else:spliter = "/"filepathsplit = filePath.split(spliter)[-2:]if len(filepathsplit) == 1:return filePathreturn filepathsplit[0] + "/" + filepathsplit[1]def autoRecognition(self):assert self.mImgList is not Noneprint("Using model from ", self.model)# uncheckedList = [i for i in self.mImgList if i not in self.fileStatedict.keys()]checkedFilePaths = [os.path.join(*Path(file).parts[-2:]) for file in self.fileStatedict if self.fileStatedict[file] == 1]uncheckedList = [file for file in self.mImgList if os.path.join(*Path(file).parts[-2:]) not in checkedFilePaths]self.autoDialog = AutoDialog(parent=self, ocr=self.ocr, mImgList=uncheckedList, lenbar=len(uncheckedList))self.autoDialog.popUp()self.currIndex = len(self.mImgList) - 1self.loadFile(self.filePath) # ADDself.haveAutoReced = Trueself.AutoRecognition.setEnabled(False)self.actions.AutoRec.setEnabled(False)self.setDirty()self.saveCacheLabel()self.init_key_list(self.Cachelabel)def reRecognition(self):img = cv2.imdecode(np.fromfile(self.filePath, dtype=np.uint8), 1)# org_box = [dic['points'] for dic in self.PPlabel[self.getImglabelidx(self.filePath)]]if self.canvas.shapes:self.result_dic = []self.result_dic_locked = ([]) # result_dic_locked stores the ocr result of self.canvas.lockedShapesrec_flag = 0for shape in self.canvas.shapes:box = [[int(p.x()), int(p.y())] for p in shape.points]kie_cls = shape.key_clsif len(box) > 4:box = self.gen_quad_from_poly(np.array(box))assert len(box) == 4# img_crop = get_rotate_crop_image(img, np.array(box, np.float32))# Do not rotatept0, pt1, pt2, pt3 = boximg_crop = img[int(pt0[1]):int(pt2[1]), int(pt0[0]):int(pt2[0])]if img_crop is None:msg = ("Can not recognise the detection box in "+ self.filePath+ ". Please change manually")QMessageBox.information(self, "Information", msg)returnresult = self.ocr.ocr(img_crop, cls=True, det=False)[0]if result[0][0] != "":if shape.line_color == DEFAULT_LOCK_COLOR:shape.label = result[0][0]result.insert(0, box)if self.kie_mode:result.append(kie_cls)self.result_dic_locked.append(result)else:result.insert(0, box)if self.kie_mode:result.append(kie_cls)self.result_dic.append(result)else:print("Can not recognise the box")if shape.line_color == DEFAULT_LOCK_COLOR:shape.label = result[0][0]if self.kie_mode:self.result_dic_locked.append([box, (self.noLabelText, 0), kie_cls])else:self.result_dic_locked.append([box, (self.noLabelText, 0)])else:if self.kie_mode:self.result_dic.append([box, (self.noLabelText, 0), kie_cls])else:self.result_dic.append([box, (self.noLabelText, 0)])try:if self.noLabelText == shape.label or result[1][0] == shape.label:print("label no change")else:rec_flag += 1except IndexError as e:print("Can not recognise the box")if (len(self.result_dic) > 0 and rec_flag > 0) or self.canvas.lockedShapes:self.canvas.isInTheSameImage = Trueself.saveFile(mode="Auto")self.loadFile(self.filePath, isAdjustScale=False)self.canvas.isInTheSameImage = Falseself.setDirty()elif len(self.result_dic) == len(self.canvas.shapes) and rec_flag == 0:if self.lang == "ch":QMessageBox.information(self, "Information", "识别结果保持一致！")else:QMessageBox.information(self, "Information", "The recognition result remains unchanged!")else:print("Can not recgonise in ", self.filePath)else:QMessageBox.information(self, "Information", "Draw a box!")def singleRerecognition(self):print("----------singleRerecognition---------------")img = cv2.imdecode(np.fromfile(self.filePath, dtype=np.uint8), 1)for shape in self.canvas.selectedShapes:box = [[int(p.x()), int(p.y())] for p in shape.points]if len(box) > 4:box = self.gen_quad_from_poly(np.array(box))assert len(box) == 4img_crop = get_rotate_crop_image(img, np.array(box, np.float32))if img_crop is None:msg = ("Can not recognise the detection box in "+ self.filePath+ ". Please change manually")QMessageBox.information(self, "Information", msg)returnresult = self.ocr.ocr(img_crop, cls=True, det=False)[0]if result[0][0] != "":result.insert(0, box)print("result in reRec is ", result)if result[1][0] == shape.label:print("label no change")else:shape.label = result[1][0]else:print("Can not recognise the box")if self.noLabelText == shape.label:print("label no change")else:shape.label = self.noLabelTextself.singleLabel(shape)self.setDirty()def TableRecognition(self):"""Table Recegnition"""from paddleocr import to_excelimport timestart = time.time()img = cv2.imdecode(np.fromfile(self.filePath, dtype=np.uint8), cv2.IMREAD_COLOR)res = self.table_ocr(img, return_ocr_result_in_table=True)TableRec_excel_dir = self.lastOpenDir + "/tableRec_excel_output/"os.makedirs(TableRec_excel_dir, exist_ok=True)filename, _ = os.path.splitext(os.path.basename(self.filePath))excel_path = TableRec_excel_dir + "{}.xlsx".format(filename)if res is None:msg = ("Can not recognise the table in "+ self.filePath+ ". Please change manually")QMessageBox.information(self, "Information", msg)to_excel("", excel_path) # create an empty excelreturn# save res# ONLY SUPPORT ONE TABLE in one imagehasTable = Falsefor region in res:if region["type"] == "table":if region["res"]["boxes"] is None:msg = ("Can not recognise the detection box in "+ self.filePath+ ". Please change manually")QMessageBox.information(self, "Information", msg)to_excel("", excel_path) # create an empty excelreturnhasTable = True# save table ocr result on PPOCRLabel# clear all old annotaions before saving resultself.itemsToShapes.clear()self.shapesToItems.clear()self.itemsToShapesbox.clear() # ADDself.shapesToItemsbox.clear()self.labelList.clear()self.indexList.clear()self.BoxList.clear()self.result_dic = []self.result_dic_locked = []shapes = []result_len = len(region["res"]["boxes"])order_index = 0for i in range(result_len):bbox = np.array(region["res"]["boxes"][i])rec_text = region["res"]["rec_res"][i][0]rext_bbox = [[bbox[0], bbox[1]],[bbox[2], bbox[1]],[bbox[2], bbox[3]],[bbox[0], bbox[3]],]# save bbox to shapeshape = Shape(label=rec_text, line_color=DEFAULT_LINE_COLOR, key_cls=None)for point in rext_bbox:x, y = point# Ensure the labels are within the bounds of the image.# If not, fix them.x, y, snapped = self.canvas.snapPointToCanvas(x, y)shape.addPoint(QPointF(x, y))shape.difficult = Falseshape.idx = order_indexorder_index += 1# shape.locked = Falseshape.close()self.addLabel(shape)shapes.append(shape)self.setDirty()self.canvas.loadShapes(shapes)# save HTML result to exceltry:to_excel(region["res"]["html"], excel_path)except Exception as e:print(f"Can not save excel file, maybe Permission denied (.xlsx is being occupied). Error: {e}")breakif not hasTable:msg = ("Can not recognise the table in "+ self.filePath+ ". Please change manually")QMessageBox.information(self, "Information", msg)to_excel("", excel_path) # create an empty excelreturn# automatically open excel annotation fileif platform.system() == "Windows":try:import win32com.clientexcept Exception as e:print("CANNOT OPEN .xlsx. It could be one of the following reasons: "f"Only support Windows | No python win32com. Error: {e}")try:xl = win32com.client.Dispatch("Excel.Application")xl.Visible = Truexl.Workbooks.Open(excel_path)# excelEx = "You need to show the excel executable at this point"# subprocess.Popen([excelEx, excel_path])# os.startfile(excel_path)except Exception as e:print("CANNOT OPEN .xlsx. It could be the following reasons: "f".xlsx is not existed. Error: {e}")else:os.system("open " + os.path.normpath(excel_path))print("time cost: ", time.time() - start)def cellreRecognition(self):"""re-recognise text in a cell"""img = cv2.imdecode(np.fromfile(self.filePath, dtype=np.uint8), cv2.IMREAD_COLOR)for shape in self.canvas.selectedShapes:box = [[int(p.x()), int(p.y())] for p in shape.points]if len(box) > 4:box = self.gen_quad_from_poly(np.array(box))assert len(box) == 4# pad around bbox for better text recognition accuracy_box = boxPad(box, img.shape, 6)img_crop = get_rotate_crop_image(img, np.array(_box, np.float32))if img_crop is None:msg = ("Can not recognise the detection box in "+ self.filePath+ ". Please change manually")QMessageBox.information(self, "Information", msg)return# merge the text result in the celltexts = ""probs = 0.0 # the probability of the cell is avgerage prob of every text box in the cellbboxes = self.ocr.ocr(img_crop, det=True, rec=False, cls=False)[0]if len(bboxes) > 0:bboxes.reverse() # top row text at firstfor _bbox in bboxes:patch = get_rotate_crop_image(img_crop, np.array(_bbox, np.float32))rec_res = self.ocr.ocr(patch, det=False, rec=True, cls=False)[0]text = rec_res[0][0]if text != "":texts += text + ("" if text[0].isalpha() else " ") # add space between english wordprobs += rec_res[0][1]probs = probs / len(bboxes)result = [(texts.strip(), probs)]if result[0][0] != "":result.insert(0, box)print("result in reRec is ", result)if result[1][0] == shape.label:print("label no change")else:shape.label = result[1][0]else:print("Can not recognise the box")if self.noLabelText == shape.label:print("label no change")else:shape.label = self.noLabelTextself.singleLabel(shape)self.setDirty()def exportJSON(self):"""export PPLabel and CSV to JSON (PubTabNet)"""import pandas as pd# automatically save annotationsself.saveFilestate()self.savePPlabel(mode="auto")# load box annotationslabeldict = {}if not os.path.exists(self.PPlabelpath):msg = "ERROR, Can not find Label.txt"QMessageBox.information(self, "Information", msg)returnelse:with open(self.PPlabelpath, "r", encoding="utf-8") as f:data = f.readlines()for each in data:file, label = each.split("\t")if label:label = label.replace("false", "False")label = label.replace("true", "True")label = label.replace("null", "None")labeldict[file] = eval(label)else:labeldict[file] = []# read table recognition outputTableRec_excel_dir = os.path.join(self.lastOpenDir, "tableRec_excel_output")# save txtfid = open("{}/gt.txt".format(self.lastOpenDir), "w", encoding="utf-8")for image_path in labeldict.keys():# load csv annotationsfilename, _ = os.path.splitext(os.path.basename(image_path))csv_path = os.path.join(TableRec_excel_dir, filename + ".xlsx")if not os.path.exists(csv_path):continueexcel = openpyxl.load_workbook(csv_path, data_only=True)sheet0 = excel.worksheets[0] # only sheet 0merged_cells = sheet0.merged_cells.ranges # list of merged cell rangeshtml_list = [["td"] * sheet0.max_column for i in range(sheet0.max_row)]for merged in merged_cells:# Convert merged cell range to start row, end row, start col, end colsr = merged.min_row - 1er = merged.max_row - 1sc = merged.min_col - 1ec = merged.max_col - 1html_list = expand_list((sr, er, sc, ec), html_list)token_list = convert_token(html_list)# load box annotationscells = []for anno in labeldict[image_path]:tokens = list(anno["transcription"])cells.append({"tokens": tokens, "bbox": anno["points"]})# 构造标注信息html = {"structure": {"tokens": token_list}, "cells": cells}d = {"filename": os.path.basename(image_path), "html": html}# 重构HTMLd["gt"] = rebuild_html_from_ppstructure_label(d)fid.write("{}\n".format(json.dumps(d, ensure_ascii=False)))# convert to PP-Structure label formatfid.close()msg = "JSON sucessfully saved in {}/gt.txt".format(self.lastOpenDir)QMessageBox.information(self, "Information", msg)def autolcm(self):vbox = QVBoxLayout()hbox = QHBoxLayout()self.panel = QLabel()self.panel.setText(self.stringBundle.getString("choseModelLg"))self.panel.setAlignment(Qt.AlignLeft)self.comboBox = QComboBox()self.comboBox.setObjectName("comboBox")self.comboBox.addItems(["Chinese & English", "English", "French", "German", "Korean", "Japanese"])vbox.addWidget(self.panel)vbox.addWidget(self.comboBox)self.dialog = QDialog()self.dialog.resize(300, 100)self.okBtn = QPushButton(self.stringBundle.getString("ok"))self.cancelBtn = QPushButton(self.stringBundle.getString("cancel"))self.okBtn.clicked.connect(self.modelChoose)self.cancelBtn.clicked.connect(self.cancel)self.dialog.setWindowTitle(self.stringBundle.getString("choseModelLg"))hbox.addWidget(self.okBtn)hbox.addWidget(self.cancelBtn)vbox.addWidget(self.panel)vbox.addLayout(hbox)self.dialog.setLayout(vbox)self.dialog.setWindowModality(Qt.ApplicationModal)self.dialog.exec_()if self.filePath:self.AutoRecognition.setEnabled(True)self.actions.AutoRec.setEnabled(True)def modelChoose(self):current_text = self.comboBox.currentText()print(current_text)lg_idx = {"Chinese & English": "ch","English": "en","French": "french","German": "german","Korean": "korean","Japanese": "japan",}if current_text in lg_idx:choose_lang = lg_idx[current_text]if hasattr(self, "ocr"):del self.ocrself.ocr = PaddleOCR(use_pdserving=False,use_angle_cls=True,det=True,cls=True,use_gpu=self.gpu,lang=choose_lang,)if choose_lang in ["ch", "en"]:if hasattr(self, "table_ocr"):del self.table_ocrself.table_ocr = PPStructure(use_pdserving=False,use_gpu=self.gpu,lang=choose_lang,layout=False,show_log=False,)else:print("Invalid language selection")self.dialog.close()def cancel(self):self.dialog.close()def loadFilestate(self, saveDir):self.fileStatepath = saveDir + "/fileState.txt"self.fileStatedict = {}if not os.path.exists(self.fileStatepath):f = open(self.fileStatepath, "w", encoding="utf-8")else:with open(self.fileStatepath, "r", encoding="utf-8") as f:states = f.readlines()for each in states:file, state = each.split("\t")self.fileStatedict[self.getImglabelidx(file)] = 1self.actions.saveLabel.setEnabled(True)self.actions.saveRec.setEnabled(True)self.actions.exportJSON.setEnabled(True)def saveFilestate(self):with open(self.fileStatepath, "w", encoding="utf-8") as f:for key in self.fileStatedict:f.write(key + "\t")f.write(str(self.fileStatedict[key]) + "\n")def loadLabelFile(self, labelpath):labeldict = {}if not os.path.exists(labelpath):f = open(labelpath, "w", encoding="utf-8")else:with open(labelpath, "r", encoding="utf-8") as f:data = f.readlines()for each in data:file, label = each.split("\t")if label:label = label.replace("false", "False")label = label.replace("true", "True")label = label.replace("null", "None")labeldict[file] = eval(label)else:labeldict[file] = []return labeldictdef savePPlabel(self, mode="Manual"):savedfile = [self.getImglabelidx(i) for i in self.fileStatedict.keys()]with open(self.PPlabelpath, "w", encoding="utf-8") as f:for key in self.PPlabel:if key in savedfile and self.PPlabel[key] != []:f.write(key + "\t")f.write(json.dumps(self.PPlabel[key], ensure_ascii=False) + "\n")if mode == "Manual":if self.lang == "ch":msg = "已将检查过的图片标签保存在 " + self.PPlabelpath + " 文件中"else:msg = "Images that have been checked are saved in " + self.PPlabelpathQMessageBox.information(self, "Information", msg)def saveCacheLabel(self):with open(self.Cachelabelpath, "w", encoding="utf-8") as f:for key in self.Cachelabel:f.write(key + "\t")f.write(json.dumps(self.Cachelabel[key], ensure_ascii=False) + "\n")def saveLabelFile(self):self.saveFilestate()self.savePPlabel()def saveRecResult(self):if {} in [self.PPlabelpath, self.PPlabel, self.fileStatedict]:QMessageBox.information(self, "Information", "Check the image first")returnbase_dir = os.path.dirname(self.PPlabelpath)rec_gt_dir = base_dir + "/rec_gt.txt"crop_img_dir = base_dir + "/crop_img/"ques_img = []if not os.path.exists(crop_img_dir):os.mkdir(crop_img_dir)with open(rec_gt_dir, "w", encoding="utf-8") as f:for key in self.fileStatedict:idx = self.getImglabelidx(key)try:img_path = os.path.dirname(base_dir) + "/" + keyimg = cv2.imdecode(np.fromfile(img_path, dtype=np.uint8), -1)for i, label in enumerate(self.PPlabel[idx]):if label["difficult"]:continueimg_crop = get_rotate_crop_image(img, np.array(label["points"], np.float32))img_name = (os.path.splitext(os.path.basename(idx))[0]+ "_crop_"+ str(i)+ ".jpg")cv2.imencode(".jpg", img_crop)[1].tofile(crop_img_dir + img_name)f.write("crop_img/" + img_name + "\t")f.write(label["transcription"] + "\n")except KeyError as e:passexcept Exception as e:ques_img.append(key)traceback.print_exc()if ques_img:QMessageBox.information(self,"Information","The following images can not be saved, please check the image path and labels.\n"+ "".join(str(i) + "\n" for i in ques_img),)QMessageBox.information(self,"Information","Cropped images have been saved in " + str(crop_img_dir),)def speedChoose(self):if self.labelDialogOption.isChecked():self.canvas.newShape.disconnect()self.canvas.newShape.connect(partial(self.newShape, True))else:self.canvas.newShape.disconnect()self.canvas.newShape.connect(partial(self.newShape, False))def autoSaveFunc(self):if self.autoSaveOption.isChecked():self.autoSaveNum = 1 # Real auto_Savetry:self.saveLabelFile()except Exception:passprint("The program will automatically save once after confirming an image")else:self.autoSaveNum = 5 # Used for backupprint("The program will automatically save once after confirming 5 images (default)")def change_box_key(self):if not self.kie_mode:returnkey_text, _ = self.keyDialog.popUp(self.key_previous_text)if key_text is None:returnself.key_previous_text = key_textfor shape in self.canvas.selectedShapes:shape.key_cls = key_textif not self.keyList.findItemsByLabel(key_text):item = self.keyList.createItemFromLabel(key_text)self.keyList.addItem(item)rgb = self._get_rgb_by_label(key_text, self.kie_mode)self.keyList.setItemLabel(item, key_text, rgb)self._update_shape_color(shape)self.keyDialog.addLabelHistory(key_text)# save changed shapeself.setDirty()def undoShapeEdit(self):self.canvas.restoreShape()self.labelList.clear()self.indexList.clear()self.BoxList.clear()self.loadShapes(self.canvas.shapes)self.actions.undo.setEnabled(self.canvas.isShapeRestorable)def loadShapes(self, shapes, replace=True):self._noSelectionSlot = Truefor shape in shapes:self.addLabel(shape)self.labelList.clearSelection()self.indexList.clearSelection()self._noSelectionSlot = Falseself.canvas.loadShapes(shapes, replace=replace)print("loadShapes") # 1def lockSelectedShape(self):"""lock the selected shapes.Add self.selectedShapes to lock self.canvas.lockedShapes,which holds the ratio of the four coordinates of the locked shapesto the width and height of the image"""width, height = self.image.width(), self.image.height()def format_shape(s):return dict(label=s.label, # strline_color=s.line_color.getRgb(),fill_color=s.fill_color.getRgb(),ratio=[[int(p.x()) / width, int(p.y()) / height] for p in s.points], # QPonitFdifficult=s.difficult, # boolkey_cls=s.key_cls, # bool)# lockif len(self.canvas.lockedShapes) == 0:for s in self.canvas.selectedShapes:s.line_color = DEFAULT_LOCK_COLORs.locked = Trueshapes = [format_shape(shape) for shape in self.canvas.selectedShapes]trans_dic = []for box in shapes:trans_dict = {"transcription": box["label"],"ratio": box["ratio"],"difficult": box["difficult"],}if self.kie_mode:trans_dict.update({"key_cls": box["key_cls"]})trans_dic.append(trans_dict)self.canvas.lockedShapes = trans_dicself.actions.save.setEnabled(True)# unlockelse:for s in self.canvas.shapes:s.line_color = DEFAULT_LINE_COLORself.canvas.lockedShapes = []self.result_dic_locked = []self.setDirty()self.actions.save.setEnabled(True)def expandSelectedShape(self):img = cv2.imdecode(np.fromfile(self.filePath, dtype=np.uint8), 1)for shape in self.canvas.selectedShapes:box = [[int(p.x()), int(p.y())] for p in shape.points]if len(box) > 4:box = self.gen_quad_from_poly(np.array(box))assert len(box) == 4box = boxPad(box, img.shape, 3)shape.points = [QPointF(box[0][0], box[0][1]),QPointF(box[1][0], box[1][1]),QPointF(box[2][0], box[2][1]),QPointF(box[3][0], box[3][1]),]print(shape.points)self.updateBoxlist()self.setDirty()def inverted(color):return QColor(*[255 - v for v in color.getRgb()])def read(filename, default=None):try:with open(filename, "rb") as f:return f.read()except Exception:return defaultdef str2bool(v):return v.lower() in ("true", "t", "1")def parse_rgb(value):r, g, b = value.split(",")r, g, b = int(r), int(g), int(b)if not (0 <= r <= 255 and 0 <= g <= 255 and 0 <= b <= 255):raise argparse.ArgumentTypeError("RGB values must be between 0 and 255.")return (r, g, b)def resolve_path(path):if getattr(sys, "frozen", False):# If the 'frozen' flag is set, we are in bundled-app mode!resolved_path = os.path.abspath(os.path.join(sys._MEIPASS, path))else:# Normal development mode. Use os.getcwd() or __file__ as appropriate in your case...resolved_path = os.path.abspath(os.path.join(os.getcwd(), path))return resolved_pathdef get_main_app(argv=[]):"""Standard boilerplate Qt application code.Do everything but app.exec_() -- so that we can test the application in one thread"""app = QApplication(argv)app.setApplicationName(__appname__)app.setWindowIcon(newIcon("app"))# Tzutalin 201705+: Accept extra arguments to change predefined class filearg_parser = argparse.ArgumentParser()arg_parser.add_argument("--lang", type=str, default="ch", nargs="?")arg_parser.add_argument("--gpu", type=str2bool, default=True, nargs="?")arg_parser.add_argument("--img_list_natural_sort", type=str2bool, default=True, nargs="?")arg_parser.add_argument("--kie", type=str2bool, default=False, nargs="?")arg_parser.add_argument("--predefined_classes_file",default=os.path.join(os.path.dirname(__file__), "data", "predefined_classes.txt"),nargs="?",)arg_parser.add_argument("--det_model_dir", type=str, default=None, nargs="?")arg_parser.add_argument("--det_limit_type", type=str, default=None, nargs="?")arg_parser.add_argument("--det_limit_side_len", type=float, default=None, nargs="?")arg_parser.add_argument("--rec_model_dir", type=str, default=None, nargs="?")arg_parser.add_argument("--rec_image_shape", type=str, default=None, nargs="?")arg_parser.add_argument("--rec_char_dict_path", type=str, default=None, nargs="?")arg_parser.add_argument("--cls_model_dir", type=str, default=None, nargs="?")arg_parser.add_argument("--bbox_auto_zoom_center", type=str2bool, default=False, nargs="?")arg_parser.add_argument("--label_font_path", type=str, default=None, nargs="?")arg_parser.add_argument("--selected_shape_color",type=parse_rgb,default="255,255,0",nargs="?",help='An RGB value as "R,G,B".',)arg_parser.set_defaults(det_model_dir=resolve_path("./custom_models/paddleocr-handwritten-digits/det/ch_PP-OCRv4_det_teacher_digits_demo_960_192"),det_limit_type="min",det_limit_side_len=192,rec_model_dir=resolve_path("./custom_models/paddleocr-handwritten-digits/rec/en_PP-OCRv3_rec_digits_demo"),rec_image_shape="3, 48, 48",rec_char_dict_path=resolve_path("./custom_models/paddleocr-handwritten-digits/rec/digits_dict.txt"))args = arg_parser.parse_args(argv[1:])win = MainWindow(lang=args.lang,gpu=args.gpu,img_list_natural_sort=args.img_list_natural_sort,kie_mode=args.kie,default_predefined_class_file=args.predefined_classes_file,det_model_dir=args.det_model_dir,det_limit_type=args.det_limit_type,det_limit_side_len=args.det_limit_side_len,rec_model_dir=args.rec_model_dir,rec_image_shape=args.rec_image_shape,rec_char_dict_path=args.rec_char_dict_path,cls_model_dir=args.cls_model_dir,bbox_auto_zoom_center=args.bbox_auto_zoom_center,label_font_path=args.label_font_path,selected_shape_color=args.selected_shape_color,)win.show()return app, windef main():"""construct main app and run it"""app, _win = get_main_app(sys.argv)return app.exec_()if __name__ == "__main__":sys.exit(main())
打包
需要在 pyinstaller 打包脚本中新增 --add-data ./custom_models:./custom_models 选项。
综合起来，形成一个 bat 脚本 build_exe.bat：
Copy "predict_system.py" "C:/Users/yong/miniconda3/envs/ppocrlabel/Lib/site-packages/paddleocr/tools/infer/predict_system.py" /y
rd /s /q build
pyinstaller -c PPOCRLabel.py --collect-all paddleocr --collect-all pyclipper --collect-all imghdr --collect-all skimage --collect-all imgaug --collect-all scipy.io --collect-all lmdb --hidden-import requests --collect-all paddle --collect-submodules shapely --hidden-import=pyqt5 -p ./libs --add-data ./libs:./libs -p ./ -p ./data --add-data ./data:./data -p ./resources --add-data ./resources:./resources --add-data ./custom_models:./custom_models --clean -F
执行脚本
.\build_exe.bat
启动
PPOCRLabel.exe --lang ch
参考资料
- https://pyinstaller.org/en/stable/usage.html
- https://paddlepaddle.github.io/PaddleOCR/latest/data_anno_synth/overview.html
- https://github.com/PFCCLab/PPOCRLabel/blob/main/README_ch.md
- https://pyinstaller.org/en/stable/spec-files.html