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