|
|
@@ -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())
|