博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
tensorflow利用预训练模型进行目标检测(一):安装tensorflow detection api
阅读量:6345 次
发布时间:2019-06-22

本文共 6462 字,大约阅读时间需要 21 分钟。

 

一、tensorflow安装

首先系统中已经安装了两个版本的tensorflow,一个是通过keras安装的, 一个是按照官网教程https://www.tensorflow.org/install/install_linux#InstallingNativePip使用Virtualenv 进行安装的,第二个在根目录下,做标记以防忘记。

安装教程:

使用 Virtualenv 进行安装请按照以下步骤使用 Virtualenv 安装 TensorFlow:发出下列其中一条命令来安装 pip 和 Virtualenv:$ sudo apt-get install python-pip python-dev python-virtualenv # for Python 2.7$ sudo apt-get install python3-pip python3-dev python-virtualenv # for Python 3.n发出下列其中一条命令来创建 Virtualenv 环境:$ virtualenv --system-site-packages targetDirectory # for Python 2.7$ virtualenv --system-site-packages -p python3 targetDirectory # for Python 3.ntargetDirectory 用于指定 Virtualenv 树的顶层目录。我们的指令假定 targetDirectory 为 ~/tensorflow,但您可以选择任何目录。通过发出下列其中一条命令激活 Virtualenv 环境:$ source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh$ source ~/tensorflow/bin/activate.csh  # csh or tcsh$ . ~/tensorflow/bin/activate.fish  # fish执行上述 source 命令后,您的提示符应该会变成如下内容:(tensorflow)$ 确保安装了 pip 8.1 或更高版本:(tensorflow)$ easy_install -U pip发出下列其中一条命令以在活动 Virtualenv 环境中安装 TensorFlow:(tensorflow)$ pip install --upgrade tensorflow      # for Python 2.7(tensorflow)$ pip3 install --upgrade tensorflow     # for Python 3.n(tensorflow)$ pip install --upgrade tensorflow-gpu  # for Python 2.7 and GPU(tensorflow)$ pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU如果上述命令执行成功,请跳过第 6 步。如果上述命令执行失败,请执行第 6 步。(可选)如果第 5 步执行失败(通常是因为您所调用的 pip 版本低于 8.1),请通过发出以下格式的命令,在活动 Virtualenv 环境中安装 TensorFlow:(tensorflow)$ pip install --upgrade tfBinaryURL   # Python 2.7(tensorflow)$ pip3 install --upgrade tfBinaryURL  # Python 3.n 其中 tfBinaryURL 表示 TensorFlow Python 软件包的网址。tfBinaryURL 的正确值取决于操作系统、Python 版本和 GPU 支持。可在此处查找适合您系统的 tfBinaryURL 值。例如,如果您要为装有 Python 3.4 的 Linux 安装仅支持 CPU 的 TensorFlow,则发出以下命令以在活动 Virtualenv 环境中安装 TensorFlow:(tensorflow)$ pip3 install --upgrade  https://download.tensorflow.google.cn/linux/cpu/tensorflow-1.8.0-cp34-cp34m-linux_x86_64.whl如果您遇到任何安装问题,请参阅常见的安装问题。后续步骤安装 TensorFlow 后,请验证安装。请注意,每次使用 TensorFlow 时,您都必须激活 Virtualenv 环境。如果 Virtualenv 环境当前未处于活动状态,请调用以下命令之一: $ source ~/tensorflow/bin/activate      # bash, sh, ksh, or zsh$ source ~/tensorflow/bin/activate.csh  # csh or tcsh在 Virtualenv 环境激活后,您就可以从这个 shell 运行 TensorFlow 程序。您的提示符将变成如下所示,这表示您的 Tensorflow 环境已处于活动状态:(tensorflow)$ 用完 TensorFlow 后,可以通过发出以下命令调用 deactivate 函数来停用环境:(tensorflow)$ deactivate 提示符将恢复为您的默认提示符(由 PS1 环境变量所定义)。卸载 TensorFlow要卸载 TensorFlow,只需移除您创建的树即可。例如:$ rm -r targetDirectory
View Code

 

二、detection api配置

参考链接:https://www.jianshu.com/p/86894ccaa407 https://www.cnblogs.com/qcloud1001/p/7677661.html  https://lijiancheng0614.github.io/2017/08/22/2017_08_22_TensorFlow-Object-Detection-API/ 

Tensorflow在更新1.0版本之后多了很多新功能,其中放出了很多用tf框架写的深度网络结构( ),大大降低了开发难度,利用现成的网络结构,无论fine-tuning还是重新训练方便了不少。最近笔者终于跑通TensorFlow Object Detection API的ssd_mobilenet_v1模型,这里记录下如何完整跑通数据准备到模型使用的整个过程,相信对自己和一些同学能有所帮助。

Object Detection API提供了5种网络结构的预训练的权重,全部是用数据集进行训练,这五种模型分别是SSD+mobilenet、SSD+inception_v2、R-FCN+resnet101、faster RCNN+resnet101、faster RCNN+inception+resnet101。各个模型的精度和计算所需时间如下。下面及介绍下如何使用Object Detection去训练自己的模型。

 

使用 git 克隆 models 部分到本地,在终端输入指令:

$ git clone https://github.com/tensorflow/models.git 使用 protoc 编译

        在 models/research 目录下的终端执行:

$ protoc object_detection/protos/*.proto --python_out=.

将 object_detection/protos/ 文件下的以 .proto 为后缀的文件编译为 .py 文件输出。

出现老问题,本服务器中安装caffe时依赖的为低版本的protoc,但是tensorflow需要高版本的protoc,在编译时报错
object_detection/protos/anchor_generator.proto:12:3: Expected "required", "optional", or "repeated".object_detection/protos/anchor_generator.proto:12:32: Missing field number.

几个查看protobuf版本信息之类的命令

查看哪些路径安装了protoc:

whereis protoc

可以看到系统中安装了两个protoc

查看默认调用的protoc是哪个:

which protoc

查看默认的protoc的版本:

protoc --version

查看pip安装的protoc的信息:(我的话就显示我在anaconda下的3.5+的版本了)

pip show protobuf

可以看到系统中存在着两个版本的protoc,现在的问题是如何在不改变默认版本protoc的情况下, 调用高版本的对tensorflow进行编译

根据https://blog.csdn.net/strangerzz/article/details/73850484,考虑新装一个protoc3.0.2版本,并软连接/usr/local/bin/protoc3.0.2,这样,当我使用protoc命令时,调用的还是2.5.0版本。当我使用protoc3.0.2命令时,调用的是protoc3.0.2版本。然后当编译其他软件时,如果需要不同的版本,在Makefile中指定具体的版本。待验证可行性

安装protoc3.0.2

进入源码所在路径(比如为$PROTO_HOME)

cd $PROTO_HOME

更换到指定版本

git checkout v3.0.2

配置编译后的输出路径(比如为$PROTOC_3_0_2)

./configure --prefix=$PROTOC_3_0_2

编译

sudo make 

sudo make install

制作软连接

ln -s $PROTOC_3_0_2/bin/protoc /usr/local/bin/protoc3.0.2

然后通过命令 protoc3.0.2 object_detection/protos/*.proto --python_out=. 进行编译

问题解决,继续下一步

 

配置环境变量

        在 .bashrc 文件中加入环境变量。首先打开 .bashrc 文件:

$ sudo gedit ~/.bashrc

然后在文件末尾加入新行:

export PYTHONPATH=$PYTHONPATH:/.../models/research:/.../models/research/slim

其中省略号所在的两个目录需要填写为 models/research 文件夹、models/research/slim 文件夹的完整目录。保存之后执行如下指令:

$ source ~/.bashrc

让改动立即生效。

 

测试是否安装成功

        在 models/research 文件下执行:

$ python/python3 object_detection/builders/model_builder_test.py

如果返回 OK,表示安装成功

不过出现问题

Traceback (most recent call last):  File "object_detection/builders/model_builder_test.py", line 23, in 
from object_detection.builders import model_builder File "/home/yanjieliu/models/models/research/object_detection/builders/model_builder.py", line 22, in
from object_detection.builders import box_predictor_builder File "/home/yanjieliu/models/models/research/object_detection/builders/box_predictor_builder.py", line 21, in
from object_detection.predictors import convolutional_box_predictor File "/home/yanjieliu/models/models/research/object_detection/predictors/convolutional_box_predictor.py", line 19, in
from object_detection.core import box_predictor File "/home/yanjieliu/models/models/research/object_detection/core/box_predictor.py", line 137, in
class KerasBoxPredictor(tf.keras.Model):AttributeError: 'module' object has no attribute 'keras'
View Code

解决方法:https://blog.csdn.net/qq_41185868/article/details/82890428

升级tensorflow到最新版本即可! pip install -U tensorflow 

升级完之后是到了tensorflow-1.11.0版本

但是之前在装keras的时候因为部分原因降到了1.2版本,现在升回来可能会出现问题。先这样吧

安装完成。

 

三、下载预训练模型

根据:https://lijiancheng0614.github.io/2017/08/22/2017_08_22_TensorFlow-Object-Detection-API/  下载预训练模型

# From tensorflow/models/object_detection/mkdir checkpointscd checkpointswget http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_11_06_2017.tar.gztar zxf ssd_mobilenet_v1_coco_11_06_2017.tar.gz

另外可到https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md下载预训练模型,将上边命令中的模型名更换一下即可

 

转载于:https://www.cnblogs.com/vactor/p/9816039.html

你可能感兴趣的文章
在开发中处理海量数据的方法 思路
查看>>
datatable 获取最大值
查看>>
sqlserver2012一直显示正在还原(Restoring)和从单用户转换成多用户模式(单用户连接中)...
查看>>
spark复习总结02
查看>>
李瑞红201771010111《第九周学习总结》
查看>>
[译]ZOOKEEPER RECIPES-Barriers
查看>>
navicat下载安装和激活一分钟完成
查看>>
6_5 一些有用网址
查看>>
NFC 鏈表操作
查看>>
pymongo模块
查看>>
第0次作业
查看>>
思维导图五个关键秘诀
查看>>
Ubuntu里设置python默认版本为python3(转载)
查看>>
快排+折半查找
查看>>
c# GC 新典型
查看>>
ssh bash 通配符
查看>>
seajs在jquery多个版本下引用jquery的插件的方案
查看>>
关于网络上java,php和.net的“口角之争“的一点想法 !
查看>>
python 第二周(第十三天) 我的python成长记 一个月搞定python数据挖掘!(21) -正则表达式re...
查看>>
[POI2011]SEJ-Strongbox
查看>>