使用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 的代碼。