faster rcnn 训练pascal_voc数据集

最新推荐文章于 2022-08-12 23:20:48 发布

原创

最新推荐文章于 2022-08-12 23:20:48 发布 · 4.1k 阅读

5 ·

CC 4.0 BY-SA版权

本文介绍了如何在Faster R-CNN框架下训练PASCAL VOC2007数据集，涵盖了硬件配置、软件安装及训练流程。

硬件需求和软件的安装和fast rcnn类似,直接开始训练voc2007。

1 下载数据集，并解压

cd py_faster_rcnn/data

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

tar xvf VOCtrainval_06-Nov-2007.tar
tar xvf VOCtest_06-Nov-2007.tar
tar xvf VOCdevkit_08-Jun-2007.tar

//建立软连接
ln -s $VOCdevkit VOCdevkit2007

//下载Imagenet预训练的模型，
cd $FRCN_ROOT
./data/scripts/fetch_imagenet_models.sh

//开始训练，数据集使用pascal_voc
cd $FRCN_ROOT
./experiments/scripts/faster_rcnn_end2end.sh [GPU_ID] [NET] [--set ...]