风之物语 发表于 2019-1-9 13:37:37

一键部署基于树莓派和tensorflow的神经网络图像识别程序



树莓派买了很久,各种小程序倒腾了半天,无非是控制个电灯开关,打开个网页,做个xmbc媒体播放器,但总是没有找到吊炸天的应用对不?


现在的树莓派3代,其实性能已经跟上来了,即使在上面跑神经网络图像识别软件,速度也勉强能接受了。本文教你如何一步完成这个神软件的部署。


谷歌2015年发布了tensorflow平台,2017年继续发布armv7版本,因此树莓派上现在能顺利跑tensorflow了。大家都知道,训练神经网络,非常耗费时间,要动用大量计算资源,还动不动得跑十几天才能完成,一般人玩不起这个。谷歌慷慨的将其在图像分类识别领域的最新训练成果Inception-v3模型的训练结果公布出来。这个模型在imageNet上面训练了几十万张共1000个不同类型的物品的图片,最终训练的错误率为3.46%(据说人眼识别的错误率仅为5.1%)。也就是说对于这1000种物品的识别率,Inception-v3模型已经高于人眼。


△ 模型识别

那就让我们在树莓派上开始部署这个程序吧!


为了让大家很方便的使用到这个程序,笔者将将所有准备工作都编译到docker镜像中去了,这样大家可以跳过繁杂的安装过程,同时笔者还略微修改了谷歌的代码,使得输出结果能翻译成中文,便于理解。


首先您要先确定您的树莓派操作系统(raspbian)的版本要至少为jessie,如果还没有升级到这个版本,请先刷到这个版本。树莓派板卡最好为pi 3,前面几代应该也能运行(笔者没有测试过),但CPU速度比较慢,运行时间会比较长。


部署的过程其实只有一行代码:

在树莓派的命令行终端上输入以下代码**** Hidden Message *****

这是一个脚本,会自动帮你把源代码下载下来,如果没有安装docker,会自动安装docker,并下载相关的docker镜像。tensorflow所需的所有组件都已经在docker中预先准备好了,您只要运行程序就行了。整个过程需要下载700M左右的数据,请耐心等待。部署完以后,将你要识别的图像放到image_recognition文件夹里面,然后运行:

cd image-recognition/

bash imgdetect.sh [你的图片文件名]
就可以啦。例如
bash imgdetect.sh demo01.jpg
demo01.jpg是一个橙子的图片,程序会返回
orange(橙色)---(Matching = 0.85452)
受限于树莓派的CPU速度,您大约需要30秒才能得到识别结果,但已经很好玩了不是?借助这个程序,您可以往下拓展,例如利用树莓派的摄像头,定时拍摄一幅照片,然后用这个程序来进行图像识别,用喇叭说出识别的结果来。这个用来给小朋友做玩具用,还是不错的。觉得本文有帮助的,被忘了给我点个赞。感兴趣的人多的话,笔者将继续发布《一行代码在树莓派上进行人脸识别》哦。

PS:
近期由于“网络安全原因”,部分读者反映有几种异常情况,统一回复如下:
1)在git clone的时候等待很久,实际是因为网络原因无法在github上抓取代码,我把项目代码共享到百度网盘 https://pan.baidu.com/s/1b3R65g ,大家可以随意下载。下载后进入image-recogintion-master 运行 bash install.sh 即可。
2)有部分用户在执行过程中,显示“docker未安装”,然后长时间等待。这种情况也是因为网络不良导致无法安装docker。建议在树莓派上反复执行:curl -sSL https://get.docker.com | sh   直到成功为止。
3)镜像下载是一个比较长的过程,建议耐心等待。镜像文件托管到dockerhub上,如果不成功,建议在网络良好的时候执行 docker pull clarkchan/rpi-tensorflow-imagenet
4) 部分用户在执行的时候反映“memory”问题,主要是前几代树莓派内存不足,无法运行程序导致的,请确认你的硬件升级到树莓派3




PS(2018-06-03):
今天重新运行了一次,发现由于网络原因,原来的安装脚本有些步骤我也不能用了。顺便修改了一下。改了几个地方:
1) docker的安装。由于官网经常失联,安装改为apt-get install docker-ce,不从官网直接下载了。
2)镜像获取的时候,改使用root权限,避免permission denied 问题。
3) 中文翻译的功能,采用简友 阿群1986 的方案,从百度翻译改为爱词霸翻译。

作者:clarkchan
來源:简书









joepluck 发表于 2022-4-1 21:43:43

大神,收下我的膝盖。

what12 发表于 2021-11-5 10:17:07

谢谢分享

eefocus_3744880 发表于 2021-5-31 16:01:48

大佬。。正阿。。。谢谢

amitof 发表于 2021-3-9 14:23:57

树莓派人工智能,学习一下~~~~~~~~~~~~~~

navy0109 发表于 2020-11-26 20:26:28

谢谢楼主分享

cat3000 发表于 2020-11-26 11:14:10

谢谢楼主分享

jal845440925 发表于 2020-9-5 14:18:31

学习学习

yidaweb 发表于 2020-9-2 00:43:16

下载来学习一下

leoworld 发表于 2020-8-21 10:33:38

这要学习学习。。。
页: [1] 2 3 4
查看完整版本: 一键部署基于树莓派和tensorflow的神经网络图像识别程序

Baidu
map