使用TensorFlow识别照片中的物体

1.环境ubuntu14.04.5

安装TensorFlow

官方文档：https://www.tensorflow.org/install/install_linux

sudo pip  install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.1-cp27-none-linux_x86_64.whl

注意：后面的网址根据不同设置而不同：

 Python 2.7  

 CPU only:  

 https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.1-cp27-none-linux_x86_64.whl

 GPU support:  

 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl

 Note that GPU support requires the NVIDIA hardware and software described in NVIDIA requirements to run TensorFlow with GPU support.  

 Python 3.4  

 CPU only:  

 https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.1-cp34-cp34m-linux_x86_64.whl

 GPU support:  

 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp34-cp34m-linux_x86_64.whl

 Note that GPU support requires the NVIDIA hardware and software described in NVIDIA requirements to run TensorFlow with GPU support.  

 Python 3.5  

 CPU only:  

 https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.1-cp35-cp35m-linux_x86_64.whl

 GPU support:  

 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-linux_x86_64.whl

 Note that GPU support requires the NVIDIA hardware and software described in NVIDIA requirements to run TensorFlow with GPU support.  

 Python 3.6  

 CPU only:  

 https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.1-cp36-cp36m-linux_x86_64.whl

 GPU support:  

 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp36-cp36m-linux_x86_64.whl

我选择了相当于：python2.7 纯CPU

自动安装无误，若有安装失败，可能是网络问题，重新执行一遍试试。

也有可能是若干依赖没有装好：先把各种依赖都安一遍

下载神经网络模型：http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz

 mkdir model

 tar -xf archive_name.tar -C model/

下载代码：https://codeload.github.com/tensorflow/models/zip/master

 unzip models-master.zip

 cd models-master/tutorials/image/imagenet/

其中的classify_image.py是我们实现物体识别的主代码

接下来做几个测试

 classify_image.py --model_dir /home/usr/model --image_file /home/usr/model/cropped_panda.jpg

 classify_image.py --model_dir /home/usr/model --image_file /home/usr/model/testkb.jpg

 classify_image.py --model_dir /home/usr/model --image_file /home/usr/model/testmous.jpg

 giant panda, panda, panda bear, coon bear, ailuropoda melanoleuca (score = 0.89107)

 computer keyboard, keypad (score = 0.64016)

 mouse, computer mouse(score = 0.89453)

巴特西

使用TensorFlow识别照片中的物体

最新文章

热门文章