重新训练R-FCN

运行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>/

猜你喜欢

转载自blog.csdn.net/sunshinezhihuo/article/details/80218718