python - Python と Caffe によるマルチプロセッシング

Question

事前トレーニング済みのカフェモデルがあり、それを使って予測しようとしています。ただし、メインプログラムの速度が低下しないように、python の multiprocessing モジュールを使用して別のプロセスでこの予測を実行しようとしています。ただし、予測を行おうとするとプロセスがフリーズするようです。これはカフェで可能ですか？そうでない場合、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 を使用してバッチ モードで画像を処理できます。これは、画像を 1 つずつ処理するよりもはるかに高速です。詳細については、pycaffe 分類器クラス https://github.com/BVLC/caffe/blob/master/python/caffe/classifier.py#L89のコードを調べることができます 。