下面以在ncnn上實現caffe網絡模型為例,和大家分享下ncnn這個牛叉的網絡框架的使用指南。
準備caffe網絡和模型
caffe 的網絡和模型通常是搞深度學習的研究者訓練出來的,一般來說訓練完會有
train.prototxt
deploy.prototxt
snapshot_10000.caffemodel
部署的時候只需要 TEST 過程,所以有 deploy.prototxt 和 caffemodel 就足夠了
alexnet 的 deploy.prototxt 可以在這里下載 https://github.com/BVLC/caffe/tree/master/models/bvlc_alexnet
alexnet 的 caffemodel 可以在這里下載 http://dl.caffe.berkeleyvision.org/bvlc_alexnet.caffemodel
轉換ncnn網絡和模型
caffe 自帶了工具可以把老版本的 caffe 網絡和模型轉換為新版(ncnn的工具只認識新版
upgrade_net_proto_text [老prototxt] [新prototxt]
upgrade_net_proto_binary [老caffemodel] [新caffemodel]
輸入層改用 Input,因為每次只需要做一個圖片,所以第一個 dim 設為 1