浏览代码

fix: 更新pipeline配置以使用新的模型并禁用适配器选项

zhch158_admin 1 周之前
父节点
当前提交
1388eb2315
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      zhch/paddleocr_vl_single_process.py

+ 3 - 2
zhch/paddleocr_vl_single_process.py

@@ -19,7 +19,8 @@ if __name__ == "__main__":
             # "input_file": "/home/ubuntu/zhch/data/至远彩色印刷工业有限公司/PaddleOCR_VL_Results/2023年度报告母公司/2023年度报告母公司_page_027.png",
             "output_dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/PaddleOCR_VL_Results",
             "collect_results": f"/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/PaddleOCR_VL_Results/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
-            "pipeline": "./my_config/PaddleOCR-VL-Client_debug.yaml",
+            "pipeline": "./my_config/PaddleOCR-VL-Client-RT-DETR-H_layout_17cls.yaml",
+            "device": "cpu",
         }
         
         # 构造参数
@@ -27,7 +28,7 @@ if __name__ == "__main__":
         for key, value in default_config.items():
             sys.argv.extend([f"--{key}", str(value)])
         # 调用PaddleOCR-VL单进程批量处理程序, 不用适配器
-        sys.argv.append("--no-adapter")
+        # sys.argv.append("--no-adapter")
         
         # 可以添加禁用标准化选项
         # sys.argv.append("--no-normalize")