python - 使用 Python 和 Caffe 进行多处理

Question

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

这是我的代码的简化版本

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 了解详细信息。