2014年12月9日火曜日

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

0 件のコメント:

コメントを投稿