diff --git a/PPOCRLabel.py b/PPOCRLabel.py index 319b5fe..b7b50c8 100644 --- a/PPOCRLabel.py +++ b/PPOCRLabel.py @@ -199,6 +199,7 @@ def get_str(str_id): "lang": self.lang, "text_detection_model_name": "PP-OCRv5_server_det", "text_recognition_model_name": "PP-OCRv5_server_rec", + "enable_mkldnn": False, } if det_model_dir is not None: @@ -210,10 +211,16 @@ def get_str(str_id): self.ocr = PaddleOCR(**params) self.text_recognizer = TextRecognition( - model_name="PP-OCRv5_server_rec", model_dir=rec_model_dir, device=self.gpu + model_name="PP-OCRv5_server_rec", + model_dir=rec_model_dir, + device=self.gpu, + enable_mkldnn=False, ) self.text_detector = TextDetection( - model_name="PP-OCRv5_server_det", model_dir=det_model_dir, device=self.gpu + model_name="PP-OCRv5_server_det", + model_dir=det_model_dir, + device=self.gpu, + enable_mkldnn=False, ) self.table_ocr = PPStructureV3( use_doc_orientation_classify=False, @@ -224,6 +231,7 @@ def get_str(str_id): use_chart_recognition=False, use_region_detection=False, device=self.gpu, + enable_mkldnn=False, ) if os.path.exists("./data/paddle.png"): @@ -3148,7 +3156,11 @@ def TableRecognition(self): ) QMessageBox.information(self, "Information", msg) # create an empty excel - tablepyxl.document_to_xl("", excel_path) + try: + tablepyxl.document_to_xl("", excel_path) + except AttributeError: # 如果 tablepyxl 报错,改用 openpyxl + wb = openpyxl.Workbook() + wb.save(excel_path) return # automatically open excel annotation file