Explorar o código

fix: 启用文档方向分类功能以改进图像处理

zhch158_admin hai 1 semana
pai
achega
66799a1716

+ 1 - 1
zhch/ppstructurev3_single_client.py

@@ -48,7 +48,7 @@ def call_api_for_image(image_path: str, api_url: str, timeout: int = 300) -> Dic
             "file": image_data,
             "fileType": 1,
             # 添加管道参数设置
-            "useDocOrientationClassify": False, # 流水分析场景关闭方向分类
+            "useDocOrientationClassify": True,
             "useDocUnwarping": False,
             "useSealRecognition": True,
             "useTableRecognition": True,

+ 2 - 2
zhch/ppstructurev3_single_process.py

@@ -109,14 +109,14 @@ def process_images_unified(image_paths: List[str],
                         predict_kwargs = {
                             'input': img_path,
                             'useLayoutDetection': kwargs.get('use_layout_detection', False),
-                            'useDocOrientationClassify': kwargs.get('use_doc_orientation', False),
+                            'useDocOrientationClassify': kwargs.get('use_doc_orientation', True),
                             'useDocUnwarping': kwargs.get('use_doc_unwarping', False),
                         }
                     else:
                         # PP-StructureV3 使用下划线命名
                         predict_kwargs = {
                             'input': img_path,
-                            'use_doc_orientation_classify': kwargs.get('use_doc_orientation', True), # 流水分析场景关闭方向分类
+                            'use_doc_orientation_classify': kwargs.get('use_doc_orientation', True),
                             'use_doc_unwarping': kwargs.get('use_doc_unwarping', False),
                             'use_layout_detection': kwargs.get('use_layout_detection', True),
                             'use_seal_recognition': kwargs.get('use_seal_recognition', True),