tf-faster-rcnn[cpu]实现目标检测(二)模型测试

本文详细介绍了如何在tf-faster-rcnn框架下使用ResNet101模型对VOC数据集进行训练的过程。从配置环境到下载预训练模型,再到数据集准备和最终的训练命令,提供了完整的步骤指导。

继CPU下运行demo之后,现在开始使用res101进行voc数据的训练
(1)在tf-faster-rcnn根目录下,执行:

NET=res101
TRAIN_IMDB=voc_2007_trainval+voc_2012_trainval
mkdir -p output/${NET}/${TRAIN_IMDB}
cd output/${NET}/${TRAIN_IMDB}
ln -s ../../../data/voc_2007_trainval+voc_2012_trainval ./default
cd ../../..

(2)下载res101模型,并建立软连接

mkdir -p data/imagenet_weights
cd data/imagenet_weights
wget -v http://download.tensorflow.org/models/resnet_v1_101_2016_08_28.tar.gz
tar -xzvf resnet_v1_101_2016_08_28.tar.gz
mv resnet_v1_101.ckpt res101.ckpt
cd ../..

(3)下载voc数据集(提取码: 3a8y),并修改名称

mv VOCdevkit VOCdevkit2007

(4)运行(运行的参数中是GPU运行的参数设置,但是在经过第一次的修改后,为gpu_id参数设置为0的话,也是可以运行的)

./experiments/scripts/test_faster_rcnn.sh 0 pascal_voc_0712 res101
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值