소스 검색

fix: 修正输出文件名格式,使用输出目录名称作为文件名

zhch158_admin 3 달 전
부모
커밋
a2fa6202f5
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      zhch/ppstructurev3_single_process.py

+ 2 - 1
zhch/ppstructurev3_single_process.py

@@ -277,7 +277,8 @@ def main():
         }
         
         # 保存最终结果
-        output_file = os.path.join(output_dir, f"OmniDocBench_Single_batch-{args.batch_size}.json")
+        output_file_name = Path(output_dir).name
+        output_file = os.path.join(output_dir, f"{output_file_name}.json")
         final_results = {
             "stats": stats,
             "results": results