FCN新手笔记，如何训练自己的模型

1、训练自己模型前需更改一下几个文件（以voc-fcn32s为例）：

 

第一个是模型权重文件的下载地址；

第二个是train.prototxt与val.prototxt文件的生成脚本；

第三个是开始模型训练脚本，主要负责载入权重、读取solver.prototxt配置文件、设置GPU、设置训练次数；

第四个是train.prototxt与val.prototxt的路径配置、模型存储位置等；

2、net.py、solve.py、solver.prototxt修改

a.  将自己的数据集分为train、val，并生成train.txt、val.txt，txt文件是记录训练数据的名称前缀；

b.  修改net.py：

载入train.txt、val.txt，路径根据自己情况来设置；

生成的train.prototxt与val.prototxt，根据自己情况设置存储位置；

c.  修改solver.prototxt文件：

train.prototxt与val.prototxt的载入路径；

模型输出位置；

d.  修改solve.py如下：

根据权重url下载权重文件，载入该位置；

设置gpu、设置solverprototxt载入位置；

3、基本设置完毕，运行solve.py，开始训练！