RRN安装

最新推荐文章于 2024-02-23 10:38:24 发布

XianyuFFF

最新推荐文章于 2024-02-23 10:38:24 发布

阅读量663

点赞数

分类专栏： tensorflow

本文链接：https://blog.csdn.net/cindy_lxy/article/details/102631106

版权

4 篇文章 0 订阅

订阅专栏

运行环境：python2.7 + cuda 8 + cudnn 5.1

框架 tensorflow 1.2.1

安装步骤：

1.需要先将TF-phrasecut-public装好，不需要安装TF-resnet 和TF-deeplab,代码中用其他（ tensorflow-deeplab-resnet）backbone替代了。

需要注意的几点：

[1]安装MS COCO API之前需要先安装cpython

pip install cpython

[2]下载的backbone预训练模型deeplab_resnet.ckpt 和 deeplab_resnet_init.ckpt 需要放置在 ./referseg_rrn/data/weights文件夹中

2.添加环境变量

export PYTHONPATH=./external/refer:./external/cocoapi/PythonAPI:./external/tensorflow-deeplab-resnet:$PYTHONPATH

3.训练

训练前需要先划分batch

python build_batches.py -d (数据集 Gref|unc|unc+|referit) -t (train|testA|val|testB|val)

参见：https://github.com/chenxi116/TF-phrasecut-public data preparation 用哪个划分哪个就行

关注