Browse Source

refactor(zhch): 优化 PPStructureV3 并行预测脚本

- 移除 GPU 限制,统一处理方式
- 添加环境变量加载,支持配置优化
- 调整并行处理参数,提高效率
zhch158_admin 3 tháng trước cách đây
mục cha
commit
b5c452019e
1 tập tin đã thay đổi với 5 bổ sung8 xóa
  1. 5 8
      zhch/ppstructurev3_parallel_predict_optimized.py

+ 5 - 8
zhch/ppstructurev3_parallel_predict_optimized.py

@@ -15,6 +15,9 @@ from tqdm import tqdm
 import threading
 import queue
 
+from dotenv import load_dotenv
+load_dotenv(override=True)
+
 class PPStructureV3ParallelPredictor:
     """
     PP-StructureV3并行预测器,支持多进程批处理
@@ -393,14 +396,8 @@ def main():
     # 并行处理参数
     batch_size = 4          # 批处理大小
     use_gpu = True          # 是否使用GPU
-    
-    # GPU限制:最多2个实例,CPU可以更多
-    if use_gpu:
-        max_workers = 2     # GPU推荐2个线程
-        use_multiprocessing = False  # GPU用线程
-    else:
-        max_workers = 4     # CPU可以用更多进程
-        use_multiprocessing = True   # CPU用进程
+    max_workers = 4     # CPU可以用更多进程
+    use_multiprocessing = False   # CPU用进程
     
     # 确保输出目录存在
     print(f"输出目录: {Path(output_dir).absolute()}")