训练 smallcorgi/Faster-RCNN_TF 模型（附ImageNet model百度云下载地址）

　　1. 下载训练、验证、测试数据和 VOCdevkit，下载地址：
　　　http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
　　　http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
　　　http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

　　2. 解压已经下载的数据至同一个文件夹下，文件夹名字修改为 VOCdevkit
　　 tar xvf VOCtrainval_06-Nov-2007.tar
　　 tar xvf VOCtest_06-Nov-2007.tar
　　 tar xvf VOCdevkit_08-Jun-2007.tar

　　　解压后的路径层次如下：
   　　$VOCdevkit/ 　　　　　　　　　　　 # development kit
　　 $VOCdevkit/VOCcode / 　　　　　　# VOC utility code
　　 $VOCdevkit/VOC2007     　　　　     # image sets, annotations, etc.
　　　# ... 等其他子目录 ...

　　3. 创建PASCAL VOC 数据库链接，可以将解压好的数据移至 $FRCN_ROOT/data 目录下
　　 cd $FRCN_ROOT/data
   　　ln -s $VOCdevkit VOCdevkit2007

　　4. 下载预训练的 ImageNet 模型，下载地址为：
　　链接：http://pan.baidu.com/s/1bEIK3c 密码：jwjc

　　5. 将下载的ImageNet模型名称改为VGG_imagenet.npy，并移至 $FRCN_ROOT/data/pretrain_model/ 目录下
　　 mv VGG_imagenet.npy $FRCN_ROOT/data/pretrain_model/VGG_imagenet.npy

　　6. 运行训练和测试的sh文件
   　　 cd $FRCN_ROOT
　　 ./experiments/scripts/faster_rcnn_end2end.sh $DEVICE $DEVICE_ID VGG16 pascal_voc

    　　其中的 $DEVICE 和 $DEVICE_ID 分别可以填 gpu 和 0，如下：
   　　 ./experiments/scripts/faster_rcnn_end2end.sh gpu 0 VGG16 pascal_voc

训练结束，执行到test的时候会出现一直等待的问题

Waiting for /home/username/Faster-RCNN_TF/output/faster_rcnn_end2end/voc_2007_trainval/VGGnet_fast_rcnn_iter_70000.ckpt to exist...

可以修改tools/test_net.py 中的while判断条件语句

while not os.path.exists(args.model) and args.wait:

改为

while not os.path.exists(args.model+'.meta') and args.wait:

巴特西

训练 smallcorgi/Faster-RCNN_TF 模型（附ImageNet model百度云下载地址）

最新文章

热门文章