caffe-ssd自带demo测试 No module named _caffe 和No module named caffe错误解决

caffe-ssd编译安装见教程:https://blog.csdn.net/qq_39703045/article/details/80156801

安装后想找一个demo测试下效果,没有选择训练数据,直接从网上下载的模型。

1、模型下载
使用的模型是VGG_VOC0712_SSD_300x300_iter_120000.caffemodel,下载完后解压,把里面的VGGNet文件夹移动到/home/mx/caffe/models/之下。

2、模型运行单张模型检测程序

$ cd /home/mx/caffe/examples # 在该目录下打开jupyter notebook

$ jupyter notebook

在jupyter notebook打开的网页中找到ssd_detect.ipynb,打开后发现是这样:


然后检查下面4个路径是否正确:

labelmap_file = 'data/VOC0712/labelmap_voc.prototxt'

model_def = 'models/VGGNet/VOC0712/SSD_300x300/deploy.prototxt'

model_weights = 'models/VGGNet/VOC0712/SSD_300x300/VGG_VOC0712_SSD_300x300_iter_120000.caffemodel'

image = caffe.io.load_image('examples/images/cat.jpg')

无错误则开始运行。

本人运行后jupyter总是卡死,发现终端有错误:

F0504 16:25:13.600085 29046 cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0)  CUDNN_STATUS_INTERNAL_ERROR

*** Check failure stack trace: ***


查询后发现是自己的电脑gpu和cudnn不匹配,cudnn5系列要求gup的计算能力最低为3.0,我的755的显卡正好3.0所以报错。

查询gpu计算能力网址:https://developer.nvidia.com/cuda-gpus

解决方案:将makefile.config的cudnn注释掉重新编译

编译步骤:

sudo make clean

make all

make runtest

编译成功,输入import caffe 报错:

File "/home/qh/caffe/python/caffe/pycaffe.py", line 13, in <module>
    from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \

ImportError: No module named _caffe


解决方案:在make runtest之前sudo make pycaffe

此处主要参考:https://cloud.tencent.com/developer/article/1010841

No module named _caffe 和No module named caffe错误解决

猜你喜欢

转载自blog.csdn.net/qq_39703045/article/details/80199148
今日推荐