Win7配置和运行TensorFlow:Object_Detection_API步骤

来源:互联网 发布:nginx代理apache php 编辑:程序博客网 时间:2024/06/05 07:06

前言

本文正式开始前,假设已安装完成tensorflow,并安装好Anaconda3-5.0.1(此版本为python3.6),python3.5亲测也可行。没有安装的同学们可以参考我的其他博客或是在网上搜索相关内容,应该都可以解决你的问题。

安装

1.从github上下载模型,下载地址:https://github.com/tensorflow/models  (下载的文件名为:models-master.zip) 


2.解压文件到磁盘指定目录,我的是E:\Tensorflow\ObjectDetection,重命名为models(此包内包括各种内容,我们所用到的object_detection文件夹在E:\Tensorflow\ObjectDetection\models\research文件夹下)

3.Protobuf 编译 
(1)从https://github.com/google/protobuf/releases下载win版的工具,即:protoc-3.4.0-win32.zip,解压到E:\Tensorflow\ObjectDetection目录下,生成:bin, include两个文件夹。
(2)将bin文件夹中的【protoc.exe】放到C:\Windows\System32文件夹下,在E:\Tensorflow\ObjectDetection\models\research文件夹下按住shift点击鼠标右键,打开命令窗口,输入protoc显示如下内容就说明可以开始编译了。

(3)Protobuf 编译,用protoc可执行文件编译目录object_detection/protos下的proto文件,生成Python文件。
protoc object_detection/protos/*.proto --python_out=.


测试

打开jupyter notebook进行测试


打开官方提供的文件:object_detection_tutorial.ipynb运行demo


在cell中选择runAll,正常的话稍等一会儿就会有结果





这里不知道是我电脑的问题还是怎么,我运行出来的结果一直都不对,只显示图片没有识别的框。后来也是看了很多博客,在下载模型那里用了其他的模型名称
# What model to download.MODEL_NAME = 'ssd_mobilenet_v1_coco_11_06_2017'#MODEL_NAME = 'faster_rcnn_resnet101_coco_11_06_2017'#MODEL_NAME = 'ssd_inception_v2_coco_11_06_2017'MODEL_FILE = MODEL_NAME + '.tar.gz'DOWNLOAD_BASE = 'http://download.tensorflow.org/models/object_detection/'

出现同样问题的可以试试用以上几个试试。

其他

1.模型选择
https://github.com/tensorflow/models/blob/master/object_detection/g3doc/detection_model_zoo.md


Speed表示速度,COCO mAP表示在COCO数据集上的平均准确率,第一个ssd_mobilenet就是我们默认使用的pre-train模型,可以都试试效果。

2.用来测试自己的图片
改变PATH_TO_TEST_IMAGES_DIR 的路径,我的如下所示。测试图片命名为image1.jpg   image2.jpg... 相应的数量for i in range(1, 3)也要根据自己的图片数量进行改变。




今天就先到这吧,基本上按照步骤操作的话,不会出太多问题。