Caffe : doc : Caffe モデル Zoo

by

Caffe : doc : Caffe モデル Zoo (翻訳/解説)
翻訳 : (株)クラスキャット セールスインフォメーション
日時 : 04/03/2017

* 本ページは、Caffe の本家サイトの document – Caffe Model Zoo を翻訳した上で
適宜、補足説明したものです：
    http://caffe.berkeleyvision.org/model_zoo.html
* ご自由にリンクを張って頂いてかまいませんが、sales-info@classcat.com までご一報いただけると嬉しいです。

 

本文

多くの研究者と技術者が異なるタスクのためにあらゆる種類のアーキテクチャとデータで Caffe モデルを作成してきました。これらのモデルは学習されて、単純な回帰から大規模なビジュアル分類、画像類似 (image similarity) のための Siamese ネットワーク、音声とロボティクス・アプリケーションまで広がる問題に適用されます。

これらのモデルを共有する手助けをするために、モデル zoo フレームワークを導入します :

  • Caffe モデル情報 (info) をパッケージングするための標準フォーマット。
  • Github Gists へ/からモデル情報を upload/download し、訓練された .caffemodel バイナリをダウンロードするためのツール。
  • モデル情報 Gists を共有するための中心的な wiki ページ。

 

訓練されたモデルをどこで取得するか

まず最初に、制限されない、創造的な利用のために BVLC-訓練 (済) モデルをバンドルしています。詳細は BVLC モデル・ライセンス を見てください。これらの各々は scripts/download_model_binary.py <dirname> を実行することでダウンロード可能です、ここで <dirname> は以下のように指定されます :

  • BVLC Reference CaffeNet in models/bvlc_reference_caffenet: ILSVRC 2012 上で訓練された AlexNet、ImageNet classification with deep convolutional neural networks by Krizhevsky et al. in NIPS 2012 で記述されているバージョンからは少し変更があります。 (Trained by Jeff Donahue @jeffdonahue)
  • BVLC AlexNet in models/bvlc_alexnet: ILSVRC 2012 上で訓練された AlexNet、ImageNet classification with deep convolutional neural networks by Krizhevsky et al. in NIPS 2012 で記述されているものとほぼ同じです。 (Trained by Evan Shelhamer @shelhamer)
  • BVLC Reference R-CNN ILSVRC-2013 in models/bvlc_reference_rcnn_ilsvrc13: Girshick et al. in CVPR 2014 で記述されている R-CNN の純粋な Caffe 実装です。(Trained by Ross Girshick @rbgirshick)
  • BVLC GoogLeNet in models/bvlc_googlenet: ILSVRC 2012 上で訓練された GoogLeNet です、Going Deeper with Convolutions by Szegedy et al. in ILSVRC 2014 で記述されているものとほぼ同じです。(Trained by Sergio Guadarrama @sguada)

Caffe ユーザによって作成されたコミュニティ・モデルは公開編集可能な wiki ページ にポストされています。これらのモデルは、citation とライセンスのようなそれぞれの著者の条件に従います。Thank you for sharing your models!

 

モデル情報 (info) フォーマット

caffe モデルは以下を含むディレクトリとして配布されます :

  • Solver/model prototxt(s)
  • readme.md は以下を含みます :
    • YAML front matter
      • このモデルを訓練するために使われた Caffe バージョン (tagged release or commit hash)。
      • [オプション] 訓練された .caffemodel のファイル URL と SHA1。
      • [オプション] github gist id.
    • モデルがどのデータの上で訓練されたか、モデリング選択、etc. の情報。
    • ライセンス情報。
  • [オプション] 他の有用なスクリプト。

モデル情報をホスティングする

GitHub Gist はモデル情報配布のために良いフォーマットです、何故ならばそれは複数のファイルを含むことができ、バージョン付けが可能で、そして in-browser なシンタックス・ハイライティングとマークダウン・レンダリングを持つからです。

scripts/upload_model_to_gist.sh はモデル・ディレクトリの非バイナリ・ファイルを Github Gist にアップロードして Gist ID を出力します。もし gist_id が既に /readme.md frontmatter の一部であるならば、既存の Gist を更新します。

アップロードをテストするためには scripts/upload_model_to_gist.sh models/bvlc_alexnet を試してください (don’t forget to delete the uploaded gist afterward)。

モデル info のダウンロードは scripts/download_model_from_gist.sh で簡単に行なえます。

訓練 (済) モデルをホスティングする

.caffemodel ファイルをどこにホストするかはユーザに依存します。私たちは BVLC 提供モデルを私たち自身のサーバにホストしています。Dropbox もまた素晴らしく動作します (ヒント: ?dl=1 が URL の最後に追加されていることを確認しましょう)。

scripts/download_model_binary.py は /readme.md frontmatter で指定される URL から .caffemodel をダウンロードして SHA1 を確認します。

 

BVLC モデル・ライセンス

原文 参照のこと。

 

以上