Caffeのmodelデータを取得してみた

モデル配布を行うmodel zooの説明はここ

./modelsには、以下のモデル情報が準備されている

drwxr-xr-x  2 root root 4096 Oct 20 20:53 bvlc_alexnet
drwxr-xr-x  2 root root 4096 Oct 20 20:53 bvlc_reference_caffenet
drwxr-xr-x  2 root root 4096 Oct 20 20:53 bvlc_reference_rcnn_ilsvrc13
drwxr-xr-x  2 root root 4096 Oct 20 20:53 finetune_flickr_style

ダウンロードには、予め
apt install python-yaml
でyamlのインストールが必要

Caffeのrootディレクトリで、model_zooの説明に従い、スクリプトを実行し、各モデルデータを取得する。

• python scripts/download_model_binary.py models/bvlc_alexnet
• python scripts/download_model_binary.py models/bvlc_reference_rcnn_ilsvrc13
• python scripts/download_model_binary.py models/bvlc_reference_caffenet
• python scripts/download_model_binary.py models/finetune_flickr_style

モデルファイルは、download_model_binary.pyで指定したフォルダーにダウンロードされる。

各モデルのファイルサイズは、以下である。

• bvlc_alexnet.caffemodel　232M
• bvlc_reference_rcnn_ilsvrc13.caffemodel 220M
• bvlc_reference_caffenet.caffemodel 232M
• finetune_flickr_style.caffemodel 217M