使用Python和Caffe进行多进程处理

Question

我有一个预训练的caffe模型，我正在尝试使用它进行预测。然而，我想在一个单独的进程中使用Python的multiprocessing模块来执行这个预测，以避免减慢主程序的速度。但是，当尝试进行预测时，该进程似乎会冻结。Caffe支持这样的操作吗？如果不支持，是否有其他方法可以在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

- user3543300

Answer 1

您可以使用pycaffe批处理模式处理图像。这比逐个处理图像要快得多。 您可以查看pycaffe分类器类的代码 https://github.com/BVLC/caffe/blob/master/python/caffe/classifier.py#L89 以获取详细信息。

- kli_nlpr

---

我通过视频流获取图像，因此无法使用批处理。是否有一种方法可以使用Python的多进程模块来完成这个任务？ - user3543300