用Python和Caffe进行多处理
我有一个预训练的caffe模型,我正试图用它做一个预测。然而,我正试图使用python的多处理模块在一个单独的过程中执行此预测,以便不会减慢主程序的速度。然而,当试图做出预测时,这个过程似乎冻结了。咖啡馆能做到这一点吗?如果没有,是否有其他方法可以在python中使用caffe进行高效的并行处理 这是我的代码的简化版本用Python和Caffe进行多处理,python,multiprocessing,caffe,pycaffe,Python,Multiprocessing,Caffe,Pycaffe,我有一个预训练的caffe模型,我正试图用它做一个预测。然而,我正试图使用python的多处理模块在一个单独的过程中执行此预测,以便不会减慢主程序的速度。然而,当试图做出预测时,这个过程似乎冻结了。咖啡馆能做到这一点吗?如果没有,是否有其他方法可以在python中使用caffe进行高效的并行处理 这是我的代码的简化版本 def predict(pretrained_network, data): prediction = pretrained_network.predict([data]
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
您可以使用pycaffe以批处理模式处理图像。这比逐个处理图像要快得多。 您可以检查pycaffe分类器类的代码
详细信息。我的图像是通过视频流输入的,因此批处理不是一个选项。有没有办法用python的多处理模块实现这一点?