Explorar o código

refactor(zhch): 调整 PP-StructureV3 单进程版参数处理逻辑

- 移除了 max_files 参数的传递
- 更新了 image_files 获取方式
- 添加了默认配置参数,包括输入和输出目录、设备选择等
- 修改了结果收集文件名,使其包含时间戳
- 注释掉了测试模式参数
zhch158_admin hai 3 meses
pai
achega
ad7d32bee6
Modificáronse 1 ficheiros con 10 adicións e 2 borrados
  1. 10 2
      zhch/ppstructurev3_single_process.py

+ 10 - 2
zhch/ppstructurev3_single_process.py

@@ -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())