运行demo_rfcn.py只是跑人家训练好的模型,接下来自己训练。(VOC数据集)
1.准备数据
下载VOC数据集:
# Download the training, validation, test data and VOCdevkit
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
2.解压缩
# Extract all of these tars into one directory named VOCdevkit
tar xvf VOCtrainval_06-Nov-2007.tar
tar xvf VOCtest_06-Nov-2007.tar
tar xvf VOCdevkit_08-Jun-2007.tar
tar xvf VOCtrainval_11-May-2012.tar
2、下载预训练好的模型
下载的模型是指在ImageNet数据集上预训练的ResNet模型,自己训练VOC数据时是在这个预训练的模型上进行fine-tuning的。
3.开始训练
cd $RFCN_ROOT
./experiments/scripts/rfcn_end2end.sh 0 ResNet-50 pascal_voc
# DATASET in {pascal_voc, coco} is the dataset to use(I only tested on pascal_voc)
# NET in {ResNet-50, ResNet-101}
output文件:
# Trained R-FCN networks are saved under:
output/<experiment directory>/<dataset name>/
# Test outputs are saved under:
output/<experiment directory>/<dataset name>/<network snapshot name>/