|
@@ -15,6 +15,9 @@ from tqdm import tqdm
|
|
|
import threading
|
|
import threading
|
|
|
import queue
|
|
import queue
|
|
|
|
|
|
|
|
|
|
+from dotenv import load_dotenv
|
|
|
|
|
+load_dotenv(override=True)
|
|
|
|
|
+
|
|
|
class PPStructureV3ParallelPredictor:
|
|
class PPStructureV3ParallelPredictor:
|
|
|
"""
|
|
"""
|
|
|
PP-StructureV3并行预测器,支持多进程批处理
|
|
PP-StructureV3并行预测器,支持多进程批处理
|
|
@@ -393,14 +396,8 @@ def main():
|
|
|
# 并行处理参数
|
|
# 并行处理参数
|
|
|
batch_size = 4 # 批处理大小
|
|
batch_size = 4 # 批处理大小
|
|
|
use_gpu = True # 是否使用GPU
|
|
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()}")
|
|
print(f"输出目录: {Path(output_dir).absolute()}")
|