CPU版SSD(Single Shot MultiBox Detector)检测单张图片

假设我们已经安装配置了CPU版SSD(Single Shot MultiBox Detector),如果你还没有,请参照我的上一篇博客

http://blog.csdn.net/gaohuazhao/article/details/72664145

我们已经安装完毕,并且跑了测试,那么怎么应用SSD实现单张图像的检测呢,让我们行动吧~

1.安装jupyter notebook

sudo pip install jupyter

2.打开jupyter notebook

jupyter notebook

到这,大家就想问了,jupyter notebook是个什么鬼,简单来说它就是一个shell,只是它比一般的shell强大了很多很多很多,至于多么多,你可以网上搜索学习一下,在这里之所以用它是因为作者的例子代码用的就是.ipynb格式。

3.打开jupyter notebook后你就会发现它真的很精彩

CPU版SSD(Single Shot MultiBox Detector)检测单张图片_第1张图片

好了,在里面找到你的caffe/examples找到ssd_detect.ipynb,打开它

3.1修改下面的caffe_root,改成你自己的路径,将import caffe下面的代码注释掉,因为我们要用cpu版的,唉

CPU版SSD(Single Shot MultiBox Detector)检测单张图片_第2张图片


3.2修改model_weights为

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

CPU版SSD(Single Shot MultiBox Detector)检测单张图片_第3张图片

确认一下四个路径是否正确

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')

imge的地方可以换成你想要检测的图片,如果没有问题,继续下一步

4.从头开始shift+enter

一步步执行到最后就会出来结果了,下面是我的结果展示

CPU版SSD(Single Shot MultiBox Detector)检测单张图片_第4张图片

CPU版SSD(Single Shot MultiBox Detector)检测单张图片_第5张图片


ok了,速度还是很快的。如有问题,欢迎交流,邮箱[email protected] QQ:965016550

感觉jupyter notebook虽然好用,但是有些操作不太方便,我们习惯直接用.py代码,它的操作栏里可以直接保存为.py文件。保存后执行,但是我的发现不显示图像,检查后在代码最后加上plt.show()就可以正常显示了

参考文章:

http://blog.csdn.net/jesse_mx/article/details/52965281

你可能感兴趣的:(caffe学习)