Win 10 安装 TensorFlow Object Detection API踩坑笔记

TensorFlow Object Detection API:

这是Google开源的基于TensorFlow目标检测框架,可以很方便的构建,训练和部署目标检测模型。

下载地址:https://github.com/tensorflow/models

Win 10 安装 TensorFlow Object Detection API踩坑笔记_第1张图片
注意选择与TensorFlow对应版本

下载完成后将model-master 解压至任意位置后重命名为:model(不重命名也是ok的)

Proctoc:

TensorFlow Object Detection API是使用protobufs来配置模型和训练参数。使用框架之前,必须编译protobuf库。

  • 下载

下载地址:https://github.com/protocolbuffers/protobuf

我安装的时候看到其他教程上写着可以用3.6但实际安装过程中不知道什么原因并不能使用,后来用我3.4版本才编译成功。
Win 10 安装 TensorFlow Object Detection API踩坑笔记_第2张图片

  • 安装:解压后把protoc.exe文件移到C:\Windows

  • 编译

可以先看一下models-master\research\object_detection\protos下都是.proto文件、protoc就是把这些文件编译成.py文件。

models-master\research下打开cmd

输入:protoc.exe object_detection/protos/*.proto --python_out=.

输入:protoc object_detection/protos/filename.proto --python_out=.

第一种是可以把所有文件全部一起编译、如果第一种编译不成功可选用第二种单独编译每个文件,改写filename部分
Win 10 安装 TensorFlow Object Detection API踩坑笔记_第3张图片
编译完成后就是这个样子
Win 10 安装 TensorFlow Object Detection API踩坑笔记_第4张图片

添加环境变量:

新建名为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

Win 10 安装 TensorFlow Object Detection API踩坑笔记_第5张图片
运行如报错:先确认相关包没问题的话试着注释掉下划线部分再运行。
Win 10 安装 TensorFlow Object Detection API踩坑笔记_第6张图片
出现这两张图就代表成功安装成功啦、撒花~~~
Win 10 安装 TensorFlow Object Detection API踩坑笔记_第7张图片

你可能感兴趣的:(学习笔记)