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"竟然给安装成功了,路径为"/usr/bin/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

相关推荐
U***49831 天前
机器学习趋势
人工智能·机器学习
lusasky1 天前
大模型混合多语言理解的原理
人工智能·神经网络·机器学习·nlp
AI即插即用1 天前
即插即用系列 | 2025 SOTA Strip R-CNN 实战解析:用于遥感目标检测的大条带卷积
人工智能·pytorch·深度学习·目标检测·计算机视觉·cnn·智慧城市
冬虫夏草19931 天前
在transformer中使用househoulder reflection(mirror transform)替代layernorm
人工智能·transformer
沛沛老爹1 天前
AI入门之GraphRAG企业级部署性能优化策略:从索引到检索的全链路提效实践
人工智能·ai·性能优化·rag·入门知识·graphrag·lightrag
FreeBuf_1 天前
突破IAM孤岛:身份安全架构为何对保护AI与非人类身份至关重要
人工智能·安全·安全架构
大千AI助手1 天前
平衡二叉树:机器学习中高效数据组织的基石
数据结构·人工智能·机器学习·二叉树·大模型·平衡二叉树·大千ai助手
IT油腻大叔1 天前
DeepSeek-多层注意力计算机制理解
python·深度学习·机器学习
z***I3941 天前
机器学习难点
人工智能·机器学习
U***e631 天前
机器学习超参数调优:GridSearch
人工智能·机器学习