|
|
@@ -314,14 +314,14 @@ if __name__ == "__main__":
|
|
|
"collect_results": "./OmniDocBench_PPStructureV3_Results/processed_files.csv",
|
|
|
}
|
|
|
|
|
|
- default_config = {
|
|
|
- "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_PPStructureV3_Results/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
|
|
|
- }
|
|
|
+ # default_config = {
|
|
|
+ # "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_PPStructureV3_Results/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
|
|
|
+ # }
|
|
|
# 构造参数
|
|
|
sys.argv = [sys.argv[0]]
|
|
|
for key, value in default_config.items():
|