前期准备工具:1、protoc-3.4.0-win32(一定要下载3.4.0的,不然后面会报错);
2、Object Detection
3、TensorFlow,有GPU的装GPU版本(网上教程很多,自己搜索),没GPU的装CPU版本(pip install tensorflow)
1、首先下载protoc工具,下载地址为:https://github.com/protocolbuffers/protobuf/releases/tag/v3.4.0
2、下载Object Detection,网址为:https://github.com/tensorflow/models
可以使用git下载,也可以直接下载zip压缩包,如下图所示:
文件夹内容如下所示:
下面打开cmd命令行,定位到F:\Tensorflow\models\research
(自己的安装路径):
然后解压下载好的protoc-3.4.0-win32,将F:\Tensorflow\protoc-3.4.0-win32\bin
(自己解压的路径)输入到cmd命令行中。protoc-3.4.0-win32文件夹bin目录中只有一个exe文件,接下来要用这个文件生成一系列的.py文件。
即输入如下命令:F:\Tensorflow\protoc-3.4.0-win32\bin\protoc object_detection/protos/*.proto --python_out=.
截图:
仔细检查路径,然后回车,如果直接切换到下一行,则没有什么问题,如果有问题,建议检查路径,因为你的可能和我的路径不一样。成功后的结果为:
那么,我们打开object_detection的文件目录(我的目录为F:\Tensorflow\models\research\object_detection
):
找到上图中的“protos"文件夹,打开它,可以看到生成了很多的.py文件(原来是没有.py文件的,只有.proto文件)
至此,就完成了90%,接下来,我们打开python的安装路径,如果是默认的话,应该就和我的路径一样:
C:\Users\HJ\AppData\Local\Programs\Python\Python36
我们在这个目录下一路定位到C:\Users\HJ\AppData\Local\Programs\Python\Python36\Lib\site-packages
这个目录下,在”site-packages“文件夹下新建一个.pth(先新建一个txt文件,然后另存即可)文件,自己可以随意取名,我的命名为”tensorflow_model.pth“,在这个文件中需要放入三个路径,分别为:
F:\Tensorflow\models\research
F:\Tensorflow\models\research\object_detection
F:\Tensorflow\models\research\slim
一定要根据自己的路径来,不要直接拷贝!!!然后保存即可。我的文件:
到这里就已经配置好了,接下来我们测试一下,打开cmd命令,定位到这个目录:
F:\Tensorflow\models\research\object_detection\builders
这个目录是object_Detection下的一个目录,好好找找就能找到。
然后输入一下命令:python model_builder_test.py
,如下图所示:
然后回车:
如果出现如上结果,则说明配置成功。没有出现的话,建议重启试试,不然就是路径不对,好好检查!
注意!
如果运行过程中遇到问题,比如,没有Cython、lxml、jupyter、matplotlib等模块,需要自行安装,可用Anaconda或直接使用cmd命令pip install xxx
安装。