Просмотр исходного кода

fix: 修改参数名称,从 'img_path' 更改为 'input',以提高代码一致性

zhch158_admin 2 недель назад
Родитель
Сommit
b2e8e04529
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      zhch/ppstructurev3_single_process.py

+ 1 - 1
zhch/ppstructurev3_single_process.py

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