|
@@ -209,7 +209,7 @@ def main():
|
|
|
return 1
|
|
return 1
|
|
|
|
|
|
|
|
print(f"Input dir: {input_dir}")
|
|
print(f"Input dir: {input_dir}")
|
|
|
- image_files = get_image_files_from_dir(input_dir, args.max_files)
|
|
|
|
|
|
|
+ image_files = get_image_files_from_dir(input_dir)
|
|
|
|
|
|
|
|
output_dir = Path(args.output_dir).resolve()
|
|
output_dir = Path(args.output_dir).resolve()
|
|
|
print(f"Output dir: {output_dir}")
|
|
print(f"Output dir: {output_dir}")
|
|
@@ -314,12 +314,20 @@ if __name__ == "__main__":
|
|
|
"collect_results": "./OmniDocBench_Results_Single/processed_files.csv",
|
|
"collect_results": "./OmniDocBench_Results_Single/processed_files.csv",
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
+ default_config = {
|
|
|
|
|
+ "input_csv": "./OmniDocBench_Results_Single/processed_files.csv",
|
|
|
|
|
+ "output_dir": "./OmniDocBench_Results_Single",
|
|
|
|
|
+ "pipeline": "PP-StructureV3",
|
|
|
|
|
+ "device": "gpu:0",
|
|
|
|
|
+ "batch_size": 2,
|
|
|
|
|
+ "collect_results": f"./OmniDocBench_Results_Single/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
|
|
|
|
|
+ }
|
|
|
# 构造参数
|
|
# 构造参数
|
|
|
sys.argv = [sys.argv[0]]
|
|
sys.argv = [sys.argv[0]]
|
|
|
for key, value in default_config.items():
|
|
for key, value in default_config.items():
|
|
|
sys.argv.extend([f"--{key}", str(value)])
|
|
sys.argv.extend([f"--{key}", str(value)])
|
|
|
|
|
|
|
|
# 测试模式
|
|
# 测试模式
|
|
|
- sys.argv.append("--test_mode")
|
|
|
|
|
|
|
+ # sys.argv.append("--test_mode")
|
|
|
|
|
|
|
|
sys.exit(main())
|
|
sys.exit(main())
|