Explorar el Código

feat(zhch): 添加单进程 DotsOCR 启动配置

- 在主程序入口处添加了启动单进程 DotsOCR 的逻辑
- 增加了命令行参数处理,支持自定义输入输出目录
- 添加了默认配置,以便在没有命令行参数时使用
- 打印 CUDA_VISIBLE_DEVICES 环境变量信息
zhch158_admin hace 3 meses
padre
commit
6d92018c5e
Se han modificado 1 ficheros con 22 adiciones y 1 borrados
  1. 22 1
      zhch/OmniDocBench_DotsOCR.py

+ 22 - 1
zhch/OmniDocBench_DotsOCR.py

@@ -8,6 +8,7 @@
 """
 
 import os
+import sys
 import json
 import tempfile
 import uuid
@@ -419,4 +420,24 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    print(f"🚀 启动单进程DotsOCR程序...")
+    print(f"🔧 CUDA_VISIBLE_DEVICES: {os.environ.get('CUDA_VISIBLE_DEVICES', 'Not set')}")
+    
+    if len(sys.argv) == 1:
+        # 如果没有命令行参数,使用默认配置运行
+        print("ℹ️  No command line arguments provided. Running with default configuration...")
+        
+        # 默认配置
+        default_config = {
+            "input_dir": "../../OmniDocBench/OpenDataLab___OmniDocBench/images",
+            "output_dir": "./OmniDocBench_Results_Single",
+        }
+        # 构造参数
+        sys.argv = [sys.argv[0]]
+        for key, value in default_config.items():
+            sys.argv.extend([f"--{key}", str(value)])
+        
+        # 测试模式
+        # sys.argv.append("--test_mode")
+    
+    sys.exit(main())