天天看点

tensorflow contrib_WSL的深度学习环境(三)Tensorflow的object_Detection安装

在Github的tensorflow官网里,可以找到Tensorflow的模型花园(Model Garden),里面有很多tensorflow的SOTA(state-of-the-art )模型和项目。https://github.com/tensorflow/models

tensorflow contrib_WSL的深度学习环境(三)Tensorflow的object_Detection安装

tensorflow的模型花园

我们只关注其中在research下的一个项目:object_detection(目标检测)。为前面安装好的WSL和Anaconda配置一个tensorflow的运行环境,能够运行起来这个目标检测项目下的Jupyter notebook:object_detection_tutorial.ipynb,网上和许多计算机视觉书籍都有关于这个目标检测代码的解释。https://github.com/tensorflow/models/tree/master/research/object_detection

tensorflow contrib_WSL的深度学习环境(三)Tensorflow的object_Detection安装

tensorflow的目标检测

通过git下载这个models目录(其实只有research目录有用)非常费时间,网络上已经有人下载好了,大家可以到百度搜索这个网盘资源,但是从Github可以下载到最新的版本(目前还在有部分更新)。

把research目录移动到home目录下,现在开始安装tensorflow。

激活前面已经创建好的tensorflow:

     source activate tensorflow
           
  • 安装1.14版本的tensorflow
     conda install tensorflow=1.14
           
  • 降级numpy到1.16

安装完成后,用conda list命令可以查看到numpy可能是高于1.16版本的1.18,只是后需要降级安装。安装命令如下:

    conda install numpy=1.16
           
  • 编译proto文件

进入到刚才下载的research目录,用protoc命令生成protos目录下proto协议文件的python文件。

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

Google Protocol Buffer( 简称 Protobuf) 是 Google 公司内部的混合语言数据标准。他们用于 RPC 系统和持续数据存储系统。提供一个具有高效的协议数据交换格式工具库(类似Json)。

但相比于Json,Protobuf有更高的转化效率,时间效率和空间效率都是JSON的3-5倍。

可用于通讯协议、数据存储等领域的语言无关、平台无关、可扩展的序列化结构数据格式。protoc是proto文件的编译工具。

  • 运行research目录下的setup.py
python setup.py buildpython setup.py install
           
  • 创建模型测试

还在reaserch目录下,运行创建模型的测试文件

python object_detection/builders/model_builder_test.py
           

这时会报一些没有找到模块的错误,一一搜索百度安装即可。

例如:ModuleNotFoundError: No module named 'matplotlib'

conda install matplotlib
           

当测试通过,会看到如图结果:

tensorflow contrib_WSL的深度学习环境(三)Tensorflow的object_Detection安装

model_builder_test通过的结果

  • 运行jupyter服务
jupyter notebook
           
tensorflow contrib_WSL的深度学习环境(三)Tensorflow的object_Detection安装

启动Jupyter服务

由于WSL没有安装可视化的图形界面,就没有浏览器可以调出,所以只能拷贝上图中的链接地址,在Windows下打开浏览器输入。看到如下的网页:

tensorflow contrib_WSL的深度学习环境(三)Tensorflow的object_Detection安装

jupyter notebook网页

在object_detection目录下找到这个.ipynb的文件,点击打开。

tensorflow contrib_WSL的深度学习环境(三)Tensorflow的object_Detection安装

Demo程序运行

点击上面的Run按钮可以单步调试运行,如果只是警告,可以得到最后运行的结果如图:

tensorflow contrib_WSL的深度学习环境(三)Tensorflow的object_Detection安装

运行结果

  • 前面可能遇到的问题:
  • 如果安装了2.0版本的tensorflow,需要卸载根除所有tensorflow版本,再重新安装
conda uninstall tensorflow
           
  • ModuleNotFoundError: No module named 'absl'
 conda install absl-py
           
  • ModuleNotFoundError: No module named 'PIL'
conda install pillow
           
  • AttributeError: module 'tensorflow' has no attribute 'contrib'

Tensorflow 需要降级, 如果是 'keras' 需要tensorflow升级