关注分享主机优惠活动
国内外VPS云服务器

如何解决在conda中安装caffe并在Linux下将pb转换为caffe的问题(ubuntu caffe安装)

本文介绍了关于“如何解决在conda中安装caffe并在Linux下从pb切换到caffe的问题”的知识。在实际案例的操作中,许多人都会遇到这样的困难,因此让主机频道带领您学习如何处理这些情况!希望你能认真阅读,有所收获!

直接从安装caffe开始:

1.首先创建一个新的python3.5环境。

conda create-n caffe-py 3.5 python = 3.c默认值输入y进行安装。

Linux下conda安装caffe与pb转caffe问题如何解决

激活环境

安装后,激活以进入环境。

源激活caffe-py3.53 .安装caffe-cpu版本或caffe-gpu版本。

Conda Install-c默认值Caffe。测试安装是否成功:

python 3进口咖啡Linux下conda安装caffe与pb转caffe问题如何解决

没有错误表示安装成功,并退出python。

退出()。安装tensorflow:

Conda install-c默认tensorflow = = 1.15.06。安装mmdnn。

pip安装mmdnnLinux下conda安装caffe与pb转caffe问题如何解决

6.我们以tf官网下载的mobilenetv1模型为例。在转换之前,我们使用netron检查pb模型的输入和输出节点名称以及输入形状:

Linux下conda安装caffe与pb转caffe问题如何解决

记住名称和输入形状

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。转换结果:

Linux下conda安装caffe与pb转caffe问题如何解决

已成功转换为caffe模式:

Linux下conda安装caffe与pb转caffe问题如何解决

未经允许不得转载:主机频道 » 如何解决在conda中安装caffe并在Linux下将pb转换为caffe的问题(ubuntu caffe安装)

评论 抢沙发

评论前必须登录!