pytorch多进程加速及代码优化方法

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