TensorFlow Object Detection API 教程

最新推荐文章于 2024-06-17 13:39:20 发布

lecea1995

最新推荐文章于 2024-06-17 13:39:20 发布

阅读量1.5k

点赞数

分类专栏： Tensorflow 文章标签： TensorFlow API 教程目标检测

本文链接：https://blog.csdn.net/lecea1995/article/details/78475454

版权

本文提供了一份详细的TensorFlow Object Detection API教程，包括安装条件、环境配置、模型训练与应用。通过实例展示了如何在PyCharm和jupyter-notebook中运行API，以及如何更换测试图片和模型，提供了不同模型的检测结果对比。

摘要由CSDN通过智能技术生成

TensorFlow Object Detection API

这个API的目的是创建一个能够在单个图像中定位和识别多个对象的精确机器学习模型。该API是在tensorflow上构造的开源框架，易于构建、训练和部署目标检测模型。

这里Anaconda下载安装就好。

安装完以后，打开Anaconda Prompt，输入清华的仓库镜像，更新包更快：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

推荐第一行命令输入两次，以便把这个镜像地址放在首位。

继续打开Anaconda Prompt，输入：

conda create -n tensorflow python=3.6

来创建python的环境。

继续在Anaconda Prompt输入：

activate tensorflow

激活环境，然后输入命令

pip install tensorflow

要在pycharm下使用tensorflow，要设置好pycharm下解释器interpreter的路径，这里也就是tensorflow的路径。

如果是虚拟的env，或anaconda的env，那就在interpreter路径里添加对应Python。

如果当前路径里没有解释器没有这个，就点击右侧add local在电脑里找。
anaconda3—>envs—>tensorflow—->python2.7（或者Python3.6，用哪个选哪个）

这里写图片描述

安装pillow, jupyter, matplotlib, lxml，通过conda install XXX 完成安装
下载tensorflow/models: https://github.com/tensorflow/models.git
下载Protobuf，在链接https://github.com/google/protobuf/releases 中找到合适的版本下载，解压后将bin文件夹中的“protoc.exe”放到C:\Windows，或者添加环境变量。调用cmd，输入protoc发现提示missing input file，证明已经可以使用了。
编译Protobuf文件:路径转到-> \models\research\object_detection, 输入：protoc *.proto –python_out=.
不报错即可。

这里写图片描述

可以看到object_detection/protos/目录下的所有*.proto都生成了对应的py文件。
- 添加环境变量PYTHONPATH
两个：

-> \models\research

-> \models\research\slim

这里写图片描述

在->models\research\object_detection文件夹下新建 ‘object_detection_tutorial.py’,内容如下，直接运行即可。

import numpy as np
import os
import six.moves.urllib as

关注

专栏目录