使用Python和Caffe进行多处理

Question

我有一个预训练的caffe模型，我试图用它进行预测。但是，我试图在使用python的多处理模块的单独进程中执行此预测，以便不会减慢主程序的速度。然而，当尝试进行预测时，该过程似乎冻结。咖啡可以吗？如果没有，有没有其他方法可以用python在python中进行高效的并行处理？

这里是我的代码

def predict(pretrained_network, data): 
    prediction = pretrained_network.predict([data]) 
    return prediction 

#Main Program 

#Network loaded here 

while True: 
    #Grab data 
    p = multiprocessing.Process(target=predict, args=(pretrained_network, data)) 
    p.daemon = True 
    p.start() 
    #Do other stuff 

Answer 1

您可以使用过程中pycaffe批处理模式下图像的简化版本。这比逐一处理图像要快得多。 有关详细信息，您可以检查pycaffe分类器类 https://github.com/BVLC/caffe/blob/master/python/caffe/classifier.py#L89 的代码。