ubuntu下FCN跑自己的训练集、测试集

一、环境介绍

主机系统ubuntu14.04,使用caffe平台，使用的python2.7

二、准备工作

1.准备自己的数据集（包括训练集、测试集）我的包含如下内容，数据集制作目前还不了解，就不介绍了。

存储在下载的FCN源码位置：/home/qiu/code/fcn.berkeleyvision.org-master/data/pascal/VOC2007。这是我的路径。

2.下载FCN源码，fcn源码在github里面下载即可。

三、更改文件

我们使用voc-fcn32s来训练我们的数据集。先看最终文件夹包含的文件，没有的文件都在fcn.brek....源码下拷贝进来就行。

1.voc-fcn32s/solve.py更改三个地方

2.train.prototxt

3.val.protptxt

4.voc_layers.py

50行改成这样

100行改成这样

113行改成这样

215行改成这样

223行往后改成这样

5.新建deploy.prototxt文件

建好这个文件，并命名后，将train.prototxt文件的内容全部拷贝过来。然后将train的第一层改为

在者就是删除最后一层layer就行了。上面训练的部分就改完了。

6.这个是源码文件下的infer.py的更改

17行改成这样

这个文件改完是可以利用voc_fcn32s测试的。