目标:优化代码,利用多进程,进行近实时预处理、网络预测及后处理:
本人尝试了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
1