上传者: 38666753
|
上传时间: 2021-12-02 17:21:54
|
文件大小: 55KB
|
文件类型: -
目标:优化代码,利用多进程,进行近实时预处理、网络预测及后处理:
本人尝试了pytorch的multiprocessing,进行多进程同步处理以上任务。
from torch.multiprocessing import Pool,Manager
为了进行各进程间的通信,使用Queue,作为数据传输载体。
manager = Manager()
input_queue = manager.Queue()
output_queue = manager.Queue()
show_queue = manager.Queue()
即将预处理进程处理后的图像放进 input_que