tensorflow/model库源码Deeplabv3+实现(一)

文章目录

    • 1.电脑配置
    • 2.基本环境配置
    • 3.clone tensorflow/models

1.电脑配置

1.ubuntu16.04
2.anaconda3+python3.6+tensorflow1.5 (版本太低了)
3.CPU 4GB

2.基本环境配置

GIthub 官方安装链接
我没有下载Jupyter notebook

DeepLab depends on the following libraries:

    Numpy
    Pillow 1.0
    tf Slim (which is included in the "tensorflow/models/research/" checkout)
    Jupyter notebook
    Matplotlib
    Tensorflow

3.clone tensorflow/models

clone官方提供的models,我用anaconda创建了一个名为tensorflow的虚拟环境,虚你环境里安装了tensorflow,所以要将下载的models移到/home/hy/anaconda3/envs/tensorflow/lib/python3.6/site-packages/tensorflow(这是我的安装路径,根据实际修改)

git clone https://github.com/tensorflow/models

将依赖库添加到PYTHONPATH中

gedit ~/.bashrc
# From tensorflow/models/research/
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
source ~/.bashrc

测试model_test.py
tensorflow/model里所有filename_test.py文件都是用来测试filename.py所需环境是否正确。
local_test.sh测试整个有关deeplab,训练VOC2012数据集代码能否正常运行,默认为Xception网络。
如果你的显卡性能不行,只能用CPU,可使用local_test_mobilenetv2.sh.

cd anaconda3/envs/tensorflow/lib/python3.6/site-packages/tensorflow/models/research/(没错,目录很长)
python deeplab/model_test.py

不出意料,出错啦
ERROR1

import tensorflow.compat.v1 as tf
ModuleNotFoundError: No module named ‘tensorflow.compat’

可能是tensorflow版本太低,解决方案
方法一:升级tensorflow到2.0
网上用的这种升级方法不管用
pip install --ignore-installed --upgrade --ignore-installed tensorflow

pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.14.0rc0-cp36-cp36m-linux_x86_64.whl 

这个是我升级tensorflow1.5版本的方法,对应python3.6,应该
只需要修改tensorflow版本和python版本(方法暂时还没有试过)
方法二:使用旧版本的models
下载链接:https://pan.baidu.com/s/1X0B4HuRmusWeSO3391xZYw 提取码: icua
ubuntu16好像没办法安装百度网盘,使用aria2进行下载ubuntu安装使用aria2下载百度网盘内容->请注意,按照参考链接的会出错,只需要把转为后台运行命令的D1中间敲个空格。但是还是没有下载成功,只能转到windows系统下载后拷过来,如该有快捷方法的请给我留言!!!
解压后放到某个目录下,例如我是放在anaconda3/envs/tensorflow/lib/python3.6/site-packages/tensorflow(感觉不必要放在虚拟环境下,要不然每次进入目录时都很长,我是配置好之后才发现的)。
然后再运行model_test.py,注意如果你用的anaconda,一定要启动虚拟环境,再运行。

然后遇到了第二个错误
ERROR2

from nets.mobilenet import mobilenet_v2
ModuleNotFoundError: No module named ‘nets’

解决方案
参考:https://blog.csdn.net/u013249853/article/details/100089276
进入slim目录,输入:

python setup.py build
 
python setup.py install

最后成功了,出现如下结果:


Ran 5 tests in 16.902s
OK

--------------------------2020.4.13 更新--------------------------------
坑坑坑!
百度网盘里下载的models,是tensorflow110版本以上的,反正tensorflow1.5版本还是太低,所以我又创建了一个虚拟环境装了tensorflow1.14!
阴差阳错,我在models/research目录下执行python setup.py build python setup.py install才解决了没有deeplab模块的错误。

你可能感兴趣的:(tensorflow)