最近在对fcn
进行复现,这里选择tensorflow
作为复现平台。
References:
源码:https://github.com/shekkizh/FCN.tensorflow
论文:https://arxiv.org/pdf/1605.06211v1.pdf
我的环境:
ubuntu 16.04
anaconda3
tensorflow 1.7.0
tf 1.7.0的版本使用指令pip install tensorflow==1.7.0
进行安装。
大概需要的python库:
matplotlib
numpy
mkl
pillow
scipy
可以使用pip install
或者conda install
进行安装。
首先使用git clone
对源码进行下载。
下载源码包含如下四个.py文件:
在之前搭建的环境中运行FCN.py文件,则开始自动下载VGG的权重参数imagenet-vgg-verydeep-19.mat
和所需要的图片数据集ADEChallengeData2016.zip
,而后进行训练。训练时全局变量mode
为“train”
,测试时改为“visualize”
:
如果嫌运行.py下载太慢,可以自己先下载权重参数和数据集,而后将imagenet-vgg-verydeep-19.mat
放到目录./Model_zoo/imagenet-vgg-verydeep-19.mat
下,将ADEChallengeData2016.zip
解压到目录./Data_zoo/MIT_SceneParsing/ADEChallengeData2016
下。下载地址:
http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-verydeep-19.mat
http://data.csail.mit.edu/places/ADEchallenge/ADEChallengeData2016.zip
RUNNING like: