|
|
@@ -307,20 +307,20 @@ if __name__ == "__main__":
|
|
|
# 默认配置
|
|
|
default_config = {
|
|
|
"input_dir": "../../OmniDocBench/OpenDataLab___OmniDocBench/images",
|
|
|
- "output_dir": "./OmniDocBench_Results_Single",
|
|
|
+ "output_dir": "./OmniDocBench_PPStructureV3_Results",
|
|
|
"pipeline": "PP-StructureV3",
|
|
|
"device": "gpu:0",
|
|
|
"batch_size": 2,
|
|
|
- "collect_results": "./OmniDocBench_Results_Single/processed_files.csv",
|
|
|
+ "collect_results": "./OmniDocBench_PPStructureV3_Results/processed_files.csv",
|
|
|
}
|
|
|
|
|
|
default_config = {
|
|
|
- "input_csv": "./OmniDocBench_Results_Single/processed_files.csv",
|
|
|
- "output_dir": "./OmniDocBench_Results_Single",
|
|
|
+ "input_csv": "./OmniDocBench_PPStructureV3_Results/processed_files.csv",
|
|
|
+ "output_dir": "./OmniDocBench_PPStructureV3_Results",
|
|
|
"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",
|
|
|
+ "collect_results": f"./OmniDocBench_PPStructureV3_Results/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
|
|
|
}
|
|
|
# 构造参数
|
|
|
sys.argv = [sys.argv[0]]
|