项目作者: ecoto

项目描述 :
MatConvNet-DAGNN to Caffe converter
高级语言: Matlab
项目地址: git://github.com/ecoto/dagnn_caffe_deploy.git
创建时间: 2017-06-18T07:25:36Z
项目社区:https://github.com/ecoto/dagnn_caffe_deploy

开源协议:

下载


dagnn_caffe_deploy

MatConvNet-DAGNN to Caffe converter

Initial EXPERIMENTAL version of a converter from a MatConvNet DagNN model to Caffe.

Only a limited subset of layer types is currently supported: dagnn.Conv, dagnn.ReLU, dagnn.Concat, dagnn.BatchNorm, dagnn.Sum, dagnn.Pooling, dagnn.LRN, dagnn.SoftMax, dagnn.Loss, dagnn.DropOut

Sucessfully tested with:

Classification tests with the converted version of imagenet-resnet-50-dag are the same than with the MSRA version of Resnet-50 caffe model.

Using:

Installation

Just copy the files in this repository within the MatConvNet folder.

Before using dagnn_caffe_deploy, remember to add MatConvNet’s “matlab” folder to the path. MatCaffe must also be available.

Sample Usage

  1. net = load('imagenet-resnet-50-dag.mat'); % Load the MatConvNet model
  2. dagnn_caffe_deploy(net,'imagenet-resnet-50-dag'); % Run the converter speciiying the model
  3. % and the prefix for the output files

The instructions above will produce the following files:

  • imagenet-resnet-50-dag.caffemodel
  • imagenet-resnet-50-dag.prototxt
  • imagenet-resnet-50-dag_mean_image.binaryproto