caffe下载预训练的CaffeNet模型遇到的问题

在测试caffe中的cpp_classification例子时,需要先下载预训练的CaffeNet模型,执行这么一句:

./scripts/download_model_binary.py models/bvlc_reference_caffenet

这里遇到了一点小问题,提示:

import yaml
ModuleNotFoundError: No module named 'yaml'

一开始移位没有安装yaml,于是使用pip安装,结果提示已经安装了yaml,原来啊,是因为caffe用的是python2,而我的ubuntu下安装的已经是python3了,于是用了命令“sudo apt install python2.7”竟然给安装成功了。由于默认python为python3,这里只能使用下面命令下载模型了:

python2.7 ./scripts/download_model_binary.py models/bvlc_reference_caffenet

不料竟然相继出现一下两个错误:

ImportError: No module named yaml

ImportError: No module named six.moves

分别使用以下命令安装就可以了:

sudo apt install python-yaml

sudo apt install python-six

你可能感兴趣的:(算法及人工智能,caffe,人工智能,深度学习)