好记性不如烂笔头, 最近用Deeplab v2跑的图像分割,现记录如下。
官方源码地址如下:https://bitbucket.org/aquariusjay/deeplab-public-ver2/overview
但是此源码只是为deeplab网络做相应变形的caffe,如果需要fine tuning微调网络,还需要准备以下文件:
txt文件:文件中有数据集的名字列表的txt文件,训练测试集列表
训练好的init.caffemodel: 针对deeplab v2,作者有已经预训练好的两个模型参数:DeepLabv2_VGG16 和DeepLabv2_ResNet101
网络结构prototxt文件: train.prototxt和solver.prototxt,分别在:DeepLabv2_VGG16 和 DeepLabv2_ResNet101
- 官网脚本文件: 三个sh文件,建议使用脚本文件,初看虽不懂,但是比python版本的运行简单很多
注:本博客只涉及脚本版本的训练
1.建立deeplab文件夹, 并下载deeplab源代码
cd ~
mkdir deeplab
cd deeplab
git clone https://bitbucket.org/aquariusjay/deeplab-public-ver2.git
2.依次建立存放设置文件夹,预测结果文件夹,数据集txt文件夹,log文件夹,model文件夹,evaluation文件夹
mkdir -p ~/deeplab/exper/voc12/config/deeplab_largeFOV
mkdir -p ~/deeplab/exper/voc12/features/labels
mkdir -p ~/deeplab/exper/voc12/features2/labels
mkdir -p ~/deeplab/exper/voc12/list
mkdir -p ~/deeplab/exper/voc12/log
mkdir -p ~/deeplab/exper/voc12/model/deeplab_largeFOV
mkdir -p ~/deeplab/exper/voc12/res
3.下载官方给的txt文件夹,以及预训练的model和网络结构文件,如上所示。
有时候可能会打不开网页,无法访问,也可以在我的资源中下载,我已经原资料打包上传,无法在官网下载就点这里
如下:
4. 把下载解压后的相应文件移动到相应文件夹
以prototxt为后缀的网络结构文件train.prototxt 、test.prototxt 以及solver.prototxt文件移动到~/deeplab/exper/voc12/config/deeplab_largeFOV
文件夹下.
unzip prototxt_and_model.zip
mv *.prototxt ~/deeplab/exper/voc12/config/deeplab_largeFOV
mv *.caffemodel ~/deeplab/exper/voc12/model/deeplab_largeFOV
unzip link.zip
cd link
mv * ~/deeplab/exper/voc12
unzip list.zip
cd list
mv * ~/deeplab/exper/voc12/list
5.数据集处理
论文中提到的pascal voc训练,其使用的数据不只是官网下载的