Browse Source

feat: 更新默认配置,修改输入输出路径及服务器地址,调整测试模式设置

zhch158_admin 1 month ago
parent
commit
ab28730b6a
1 changed files with 6 additions and 5 deletions
  1. 6 5
      zhch/mineru2_vllm_multthreads.py

+ 6 - 5
zhch/mineru2_vllm_multthreads.py

@@ -623,10 +623,11 @@ if __name__ == "__main__":
         
         # 默认配置
         default_config = {
-            "input_file": "/home/ubuntu/zhch/data/至远彩色印刷工业有限公司/2023年度报告母公司.img/2023年度报告母公司_page_003.png",
-            "output_dir": "./output",
-            "collect_results": f"./output/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
-            "server_url": "http://127.0.0.1:8121",
+            # "input_file": "/home/ubuntu/zhch/data/至远彩色印刷工业有限公司/2023年度报告母公司.img/2023年度报告母公司_page_003.png",
+            "inout_dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水.img",
+            "output_dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/mineru-vlm-2.5.3_Results",
+            # "collect_results": f"./output/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
+            "server_url": "http://10.192.72.11:8121",
             "timeout": "300",
             "batch_size": "1",
             "dpi": "200",
@@ -638,7 +639,7 @@ if __name__ == "__main__":
             sys.argv.extend([f"--{key}", str(value)])
         
         # 测试模式和调试模式
-        sys.argv.append("--test_mode")
+        # sys.argv.append("--test_mode")
         sys.argv.append("--debug")
     
     sys.exit(main())