Object Detection API(1)——安装及测试

Object Detection API1)——安装及测试:

最近在学习Tensorflow推出的object detection api,所以写下一些笔记来记录执行的步骤以及遇到的问题,以备忘。

整个笔记过程会从object detection api安装,测试,到训练自己定义的数据,输出训练模型,以及如何用opencv来调用训练后的模型。

1,下面记录object detection api工具的安装过程,安装环境为win10

准备工作:

(1)预先安装tensorflow-gpu-1.4, python3.5

(2)下载TensorFlow/model文件夹

(3)在https://github.com/google/protobuf/releases下载 

protoc-3.4.0-win32.zip,安装protoc,设置环境变量,或者将bin和include两个文件夹,移到C:\Windows目录下。

(4)在object_detection的父目录执行命令:

protoc.exe object_detection/protos/*.proto –python_out=.

安装过程:

(1)在models-master/research目录寻找setup.py文档。若存在,则运行

python setup.py install

(2)配置环境变量,如下YOUPATH表示你的models所在目录:

YOUPATH\models-master\research:YOUPATH\models-master\research\object_detection

(3)运行测试命令:python object_detection/builders/model_builder_test.py

如果运行测试命令报错如下:

Import erro : no models named  ‘nets’

《Object Detection API(1)——安装及测试》

则需要删除slim目录的BUILD文件重新安装,在models-master/research/slim目录下也运行

python setup.py install

(4)再次测试结果如下所示:

《Object Detection API(1)——安装及测试》

2,运行官方demo

打开object_detection目录下的object_detection_tutorial.ipynb,命令如下:

jupyter notebookobject_detection_tutorial.ipynb

检测结果如下:

《Object Detection API(1)——安装及测试》

参考资料

[1]https://zhuanlan.zhihu.com/p/29215867

[2]TensorFlow model 官方开源网址:https://github.com/tensorflow/models/tree/master/object_detection

    原文作者:鸿鶱
    原文地址: https://www.jianshu.com/p/1ed2d9ce6a88
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞