RRN安装

paper：2018 CVPR  Referring Image Segmentation via Recurrent Refinement Networks

源码：https://github.com/liruiyu/referseg_rrn

运行环境：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 用哪个划分哪个就行

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！