参考文档

  1. https://caffe2.ai/doxygen-c/html/classes.html Caffe2官方C++ API在线手册
  2. Caffe:CPU模式下使用openblas-openmp(多线程版本)
  3. https://github.com/shicai/MobileNet-Caffe Caffe版本MobileNet_v1/2模型下载地址
  4. https://caffe2.ai/docs/caffe-migration.html#null__converting-models-from-caffe-to-caffe2 从Caffe迁移到Caffe2的模型转换

MobileNet_v1模型测试

从参考文档3下载mobilent模型描述文件mobilenet_deploy.prototxt和预训练模型文件mobilenet.caffemodel

合并Caffe层

因为Caffe2不支持在BatchNorm层之后接Scale层，所以从Caffe模型转换到Caffe2模型的过程中就会出现错误，显示找不到SpatialBN operator。具体可以参考这里。解决方法是合并Caffe层，具体脚本如下：

cd /opt/caffe/python/
sudo python -m gen_merged_model --output_model mobilenet_deploy_merge.prototxt  \
    --output_weights mobilenet_merge.caffemodel \
    ../models/mobilenet/mobilenet_deploy.prototxt \
    ../models/mobilenet/mobilenet.caffemodel

TIPS：在运行上述脚本之前需要先安装好pycaffe环境

sudo python -m caffe2.python.caffe_translator --init_net mobilenet_v2_init_net.pb \
    --predict_net mobilenet_v2_predict_net.pb \
    models/mobilenet_v2/mobilenet_v2_deploy.prototxt \
    models/mobilenet_v2/mobilenet_v2.caffemodel

results matching ""

    No results matching ""