本文介绍了关于“如何解决在conda中安装caffe并在Linux下从pb切换到caffe的问题”的知识。在实际案例的操作中,许多人都会遇到这样的困难,因此让主机频道带领您学习如何处理这些情况!希望你能认真阅读,有所收获!
直接从安装caffe开始:
1.首先创建一个新的python3.5环境。
conda create-n caffe-py 3.5 python = 3.c默认值输入y进行安装。
激活环境
安装后,激活以进入环境。
源激活caffe-py3.53 .安装caffe-cpu版本或caffe-gpu版本。
Conda Install-c默认值Caffe。测试安装是否成功:
python 3进口咖啡
没有错误表示安装成功,并退出python。
退出()。安装tensorflow:
Conda install-c默认tensorflow = = 1.15.06。安装mmdnn。
pip安装mmdnn
6.我们以tf官网下载的mobilenetv1模型为例。在转换之前,我们使用netron检查pb模型的输入和输出节点名称以及输入形状:
记住名称和输入形状
7.进行Pb-》Caffe转换,命令行输入:
mm convert-SF Tensor Flow-iwmobilenet _ v1 . p B- in nodename input-input shape 192,192,dst nodename mobilenet v1/predictions/reshape _ dfcaffe-omtf _ mobilenet 8。转换结果:
已成功转换为caffe模式:
评论前必须登录!
注册