Browse Source

Merge branch 'master' of https://github.com/Tencent/ncnn

tags/20180314
nihuini 8 years ago
parent
commit
c1e93152eb
1 changed files with 7 additions and 0 deletions
  1. +7
    -0
      tools/caffe/caffe2ncnn.cpp

+ 7
- 0
tools/caffe/caffe2ncnn.cpp View File

@@ -766,6 +766,13 @@ int main(int argc, char** argv)
fprintf(pp, " 1=%d", memory_data_param.height());
fprintf(pp, " 2=%d", memory_data_param.channels());
}
else if (layer.type() == "MVN")
{
const caffe::MVNParameter& mvn_param = layer.mvn_param();
fprintf(pp, " 0=%d", mvn_param.normalize_variance());
fprintf(pp, " 1=%d", mvn_param.across_channels());
fprintf(pp, " 2=%f", mvn_param.eps());
}
else if (layer.type() == "Normalize")
{
const caffe::LayerParameter& binlayer = net.layer(netidx);


Loading…
Cancel
Save