本文主要介绍根据github tensorflow/models中官方代码来训练deeplab v3+
源代码: https://github.com/tensorflow/models/tree/master/research/deeplab
配置deeplab v3
- Clone 源代码, https://github.com/tensorflow/models.git
- 根据官方文档进行安装,https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/installation.md
这里有几个需要注意的地方:
(1) cuda 9.0 & tensorflow 1.6以上版本
(2) 需要将models/research/slim路径导入到PYTHONPATH环境变量中,这个是因为deeplab中的一些工具比如multigrid使用的是slim中实现的
export PYTHONPATH=$PYTHONPATH:/path-to/models/research/slim
(3) 然后进行测试就可以得到一个结果,测试需要在models/research/下进行,这样很不方便,想要在models/research/deeplab下进行,可以在m