Caffe2-事前トレーニング済みモデルへのアクセスの確認

事前にトレーニングされたモデルをPythonアプリケーションで使用する方法を学ぶ前に、まずモデルがマシンにインストールされており、Pythonコードからアクセスできることを確認しましょう。

Caffe2をインストールすると、事前にトレーニングされたモデルがインストールフォルダーにコピーされます。Anacondaがインストールされているマシンでは、これらのモデルは次のフォルダーにあります。

anaconda3/lib/python3.7/site-packages/caffe2/python/models

これらのモデルの存在については、マシンのインストールフォルダを確認してください。次の短いPythonスクリプトを使用して、インストールフォルダーからこれらのモデルをロードしてみてください。

CAFFE_MODELS = os.path.expanduser("/anaconda3/lib/python3.7/site-packages/caffe2/python/models")
INIT_NET = os.path.join(CAFFE_MODELS, 'squeezenet', 'init_net.pb')
PREDICT_NET = os.path.join(CAFFE_MODELS, 'squeezenet', 'predict_net.pb')
print(INIT_NET)
print(PREDICT_NET)

スクリプトが正常に実行されると、次の出力が表示されます-

/anaconda3/lib/python3.7/site-packages/caffe2/python/models/squeezenet/init_net.pb
/anaconda3/lib/python3.7/site-packages/caffe2/python/models/squeezenet/predict_net.pb

これにより、 squeezenet モジュールはマシンにインストールされており、コードからアクセスできます。

これで、Caffe2を使用して画像分類用の独自のPythonコードを作成する準備が整いました。 squeezenet 事前にトレーニングされたモジュール。