Windows下安装TensorFlow Object Detection API,训练自己的数据集

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/chuquanchang1051/article/details/79804965

1.首先安装配置好TensorFlow,点environments 选择all ,输入tensorflow,进行安装
这里写图片描述
然后选择spyder
在右侧输入 import tensorflow 如果没有报错则安装成功
这里写图片描述
2.下载TensorFlow模型源码,这里写链接内容
3.编译Protobuf,生产py文件。
需要先安装Google的protobuf,下载protoc-3.4.0-win32.zip ,3.5版本很有可能会出错
打开cmd窗口,cd到models/research/目录下(老版本没有research目录),执行如下:
protoc object_detection/protos/*.proto --python_out=.
将生成一堆python文件,如下图所示:这里写图片描述
备注:
这里写图片描述
4.测试安装(建议在Anaconda prompt 下进行)

python object_detection/builders/model_builder_test.py
```![这里写图片描述](https://img-blog.csdn.net/20180403161629920?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvY2h1cXVhbmNoYW5nMTA1MQ==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast)
坑: Windows下会出现找不到包的问题:

Traceback (most recent call last):
  File "object_detection/builders/model_builder_test.py", line 21, in <module>
    from object_detection.builders import model_builder
ImportError: No module named 'object_detection'
然后并没有神马卵用,为了一劳永逸,我直接将整两个目录添加到python默认的搜索路径下就行了。 
解决方法:在site-packages添加一个路径文件,如tensorflow_model.pth,必须以.pth为后缀,写上你要加入的模块文件所在的目录名称就是了,如下图: 
![这里写图片描述](https://img-blog.csdn.net/20180403161743261?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvY2h1cXVhbmNoYW5nMTA1MQ==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast)
注意:也可直接打开文件model_builder_test.py,
在程序的一开始添加
`
import sys
sys.path.append('C:\\Users\\daiwen\\Anaconda3\\Lib\\site-packages\\tensorflow\\models\\research')
`
然后程序跳转到报错找不到nets文件 如果报错:ImportError: No module named nets
将reserch/slim目录下的BUILD删除,然后运行该文件夹里面的setup.py





<div class="se-preview-section-delimiter"></div>

这里写代码片
“`

python setup.py build
python setup.py install 

这里写图片描述
终于成功了
这里写图片描述

猜你喜欢

转载自blog.csdn.net/chuquanchang1051/article/details/79804965
今日推荐