1、首先安装配置好TensorFlow,参考地址
2、下载TensorFlow模型源码,https://github.com/tensorflow/models
(注:最好不要在C盘下使用,可能存在各种权限问题)
3、通过pip安装:pillow, jupyter, matplotlib, lxml,如下:
pip install pillow
4、编译Protobuf,生产py文件。
需要先安装Google的protobuf,下载protoc-3.4.0-win32.zip
打开cmd窗口,cd到models/research/目录下(老版本没有research目录),执行如下:
protoc object_detection/protos/*.proto --python_out=.
将生成一堆python文件,如下图所示:
an
5、为依赖包添加环境变量 (这一步很关键)
PYTHONPATH C:\tensorflow\models\research;C:\tensorflow\models\research\slim
6.然后并没有神马卵用,为了一劳永逸,我直接将整两个目录添加到python默认的搜索路径下就行了。
解决方法:在site-packages添加一个路径文件,如tensorflow_model.pth,必须以.pth为后缀,写上你要加入的模块文件所在的目录名称就是了,如下图:
测试是否安装成功:输入
python object_detection/builders/model_builder_test.py
Windows下TensorFlow Object Detection API 环境搭建时遇到的坑
最新推荐文章于 2022-03-09 22:43:37 发布