ParseNet源码训练和分析

ParseNet

• 官方源代码位置:
  Caffe: https://github.com/weiliu89/caffe/tree/fcn
  model: https://gist.github.com/weiliu89/45e9e8de2c13af6476ca#file-readme-md

编译Caffe

直接从github上clone下来代码

git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout fcn

注意：记得切换到fcn分支，不然会出现eval_type未定义的问题
存在的问题：cudnn6.0无法使用，于是在编译的时候关掉了cudnn.

训练

先制作数据集——训练集增强的VOC2012，得到完整的数据集后，不需要进行resize，因为parsenet的batch-size设置的是1, iter_size设置为8.
训练集数目为10582，验证集为1449，手动转化为lmdb，进行训练即可.
其中训练集的收集在我的deeplab中有介绍.

训练过程中显存占用为2