Win 10 安装 TensorFlow Object Detection API踩坑笔记
TensorFlow Object Detection API:
这是Google开源的基于TensorFlow目标检测框架,可以很方便的构建,训练和部署目标检测模型。
注意选择与TensorFlow对应版本
下载完成后将model-master
解压至任意位置后重命名为:model
(不重命名也是ok的)
Proctoc:
TensorFlow Object Detection API是使用protobufs来配置模型和训练参数。使用框架之前,必须编译protobuf库。
- 下载
我安装的时候看到其他教程上写着可以用3.6但实际安装过程中不知道什么原因并不能使用,后来用我3.4版本才编译成功。
-
安装:解压后把
protoc.exe
文件移到C:\Windows
-
编译:
可以先看一下models-master\research\object_detection\protos
下都是.proto文件、protoc就是把这些文件编译成.py文件。
在models-master\research\object_detection
下打开cmd
输入:protoc.exe object_detection/protos/*.proto --python_out=.
或
输入:protoc object_detection/protos/filename.proto --python_out=.
第一种是可以把所有文件全部一起编译、如果第一种编译不成功可选用第二种单独编译每个文件,改写filename
部分
编译完成后就是这个样子
添加环境变量:
新建名为xxx变量、添加E:\models-master\research\slim
(变量名可随意、路径要写自己文件的路径)
运行setup. py:
在models-master\research
下打开cmd:
输入:python setup.py install
测试安装是否成功:
在models-master\research\object_detection\builders
下打开cmd:
输入:python model_builder_test.py
运行官方示例:
在models-master\research\object_detection
下打开cmd:
输入:jupyter notebook
打开:object_detection_tutorial.ipynb
运行如报错:先确认相关包没问题的话试着注释掉下划线部分再运行。
出现这两张图就代表成功安装成功啦、撒花~~~