Преглед изворни кода

fix: 更新输出目录和结果收集路径,确保与PP-StructureV3一致

zhch158_admin пре 2 месеци
родитељ
комит
450ab8c1ef
1 измењених фајлова са 5 додато и 5 уклоњено
  1. 5 5
      zhch/ppstructurev3_single_process.py

+ 5 - 5
zhch/ppstructurev3_single_process.py

@@ -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]]