Win10环境下用Deeplabv3+模型训练Mapillary Vistas dataset(一)

最近在做街景图象的语义分割,阅读了一些文献之后准备开始实战。用的模型是这两年很火的deeplabv3+,数据集是提前申请了mapillary提供给researcher的数据集。

第一步是从GitHub下载模型

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

下载了好久,网络一直很慢,于是我换了个方法,先在码云上新建了一个仓库,然后之间把https://github.com/tensorflow/models导入这个新建的仓库。
再git clone码云上的这个链接,速度噌噌就上去了!

第二步添加依赖库到PYTHONPATH

将 deeplab 文件夹下的 train.py, eval.py, vis.py, export_model.py 和model_test.py这5个脚本的开头加以下语句
import sys sys.path.append(r'C:\Users\wangrushan\model\research') sys.path.append(r'C:\Users\wangrushan\model\research\slim')
注意必须加在 import deeplab 语句之前

第三步测试model_test.py

这里我遇到了一个error
Win10环境下用Deeplabv3+模型训练Mapillary Vistas dataset(一)_第1张图片
我先试了一下重启spyder,并没有解决这个问题。所以应该是我电脑配置的问题,gpu性能不够而数据集过大,那就只能从数据集来入手解决
我又改用anaconda prompt运行model_test.py。可惜电脑硬件是硬伤!一直卡在这里Win10环境下用Deeplabv3+模型训练Mapillary Vistas dataset(一)_第2张图片
最终还是error了
在这里插入图片描述
在听取了老师的建议后我决定曲线救国,换用cpu版本的tensorflow进行实验。

第四步将TensorFlow由gpu版切换为cpu版

这一步我直接在测试代码的所有import语句之前加上了

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

我又运行了一遍model_test.py这次终于成功了在这里插入图片描述
至此deeplab环境配置成功

未完待续。。。

你可能感兴趣的:(tensorflow)