deeplab源码:models/research/deeplab at master · tensorflow/models · GitHub

在本地仓库克隆下来即可。

注意!该源码是tensorflow1写的,装tensorflow2的话会需要改很多函数,建议用1!

可以在官网查看对应版本:在 Windows 环境中从源代码构建  |  TensorFlow

我选的是python3.7+tensorflow1.15+cuda10+cudnn7.4的组合。

1. 在anaconda中创建环境,进入环境,安装以下包。

(jupyter在python3.7下用conda安装会报错,但是在3.6下不会)

>> conda create -n tf1_deeplab python=3.7
>> activate tf1_deeplab

>> conda install numpy
>> conda install pillow
>> pip install jupyter
>> conda install matplotlib
>> pip install tensorflow-gpu==1.15.0

2.安装对应版本的cuda和cudnn

cuda:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

(一路默认安装就行了)


cudnn: https://developer.nvidia.com/rdp/cudnn-archive

 (下载下来是一个压缩包,把对应的文件放到cuda的相应的bin等目录下。)

 3.测试tf是否安装成功,若返回true则成功

>> import tensorflow as tf
>> tf.test.is_gpu_available()

4.在pycharm里选择配置好的环境'tf1_deeplab'。

运行model_test.py测试。

报错:ModuleNotFoundError: No module named ‘tf_slim‘

解决:安装tf_slim

 5.再次运行model_test.py,成功!

 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐