tf-faster-rcnn指南（一）——序言
tf-faster-rcnn指南（二）——平台搭建
tf-faster-rcnn指南（三）——训练模型
tf-faster-rcnn指南（四）——制作PASCAL VOC2007数据集
tf-faster-rcnn指南（五）——训练自己的数据集
tf-faster-rcnn指南（六）——绘制Precision-recall曲线

一、下载预训练模型
1、在data目录下创建一个imagenet_weights文件夹

mkdir -p data/imagenet_weights
cd data/imagenet_weights

2、下载预训练模型，解压到imagenet_weights文件夹下，并把解压后的权重数据重命名，因为后面在调用权重数据的时候名字需要对应的上。
　　一共有4种预训练模型可供选择

2.1 vgg16

wget -v http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz
tar -xzvf vgg_16_2016_08_28.tar.gz
mv vgg_16.ckpt vgg16.ckpt
cd ../..

2.2 res50

wget -v http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz
tar -xzvf resnet_v1_50_2016_08_28.tar.gz
mv resnet_v1_50.ckpt res50.ckpt
cd ../..

2.3 res101

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 ../..

2.4 res152
在下载预训练模型前还需要将res152.yml文件存放到/tf-faster-rcnn/experiments/cfgs/文件夹下
res152.yml下载链接：https://pan.baidu.com/s/1m5jewlZuHxdPANcBcA0LwA
提取码：kkp3

wget -v http://download.tensorflow.org/models/resnet_v1_152_2016_08_28.tar.gz
tar -xzvf resnet_v1_152_2016_08_28.tar.gz
mv resnet_v1_152.ckpt res152.ckpt
cd ../..

除此之外可以登录官方网址下载：https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models

二、训练模型
1、为了节省时间并排除错误，我把迭代次数只设置了200次,
具体操作为：
　　./experiments/scripts/train_faster_rcnn.sh里的第22行把ITERS=70000改成ITERS=200，
　　同时把./experiments/scripts/test_faster_rcnn.sh的ITERS也改成200。
　　注意：因为我使用的是pascal_voc数据集，所以只需要更改对应数据集的ITERS的就行了，训练和测试的都要改，因为在train_faster_rcnn.sh的末尾会执行test_faster_rcnn.sh。
2、开始训练(这里以vgg16为例，最后的vgg16就是对应的权重数据，名字要对的上，01是GPU的ID，pascal_voc是训练使用的数据集)：

./experiments/scripts/train_faster_rcnn.sh 01 pascal_voc vgg16

三、运行结果
1、训练loss

2、训练结果

可以看到，结果都很差，因为就迭代了200轮

tf-faster-rcnn指南（三）——训练模型相关推荐

毕设日志——配置服务器与测试gpu版本的tf faster rcnn代码
4号拿到服务器的账号密码,先开始折腾服务器怎么用. 2019.4.4 下载DokanSetup1.0.5和WinSSHFS1.6.1.13 安装打开WinSSHFS 取名字Drive Name对应本 ...
目标检测——Faster R-CNN论文阅读
论文阅读--Faster R-CNN:Towards Real-Time Object Detection with Region Proposal Networks 文章目录论文阅读--Faste ...
Faster R-CNN改进篇（一）： ION ● HyperNet ● MS CNN
版权声明:本文为博主原创文章,未经博主允许不得转载. https://blog.csdn.net/linolzhang/article/details/74159463 一. 源起于Faster 深度 ...
Faster R-CNN 深入理解改进方法汇总
Faster R-CNN 从2015年底至今已经有接近两年了,但依旧还是Object Detection领域的主流框架之一,虽然推出了后续 R-FCN,Mask R-CNN 等改进框架,但基本结构变化 ...
Faster R-CNN论文详解
原文链接:http://lib.csdn.net/article/deeplearning/46182 paper链接:Faster R-CNN: Towards Real-Time Object D ...
[目标检测] Faster R-CNN 深入理解改进方法汇总
Faster R-CNN 从2015年底至今已经有接近两年了,但依旧还是Object Detection领域的主流框架之一,虽然推出了后续 R-FCN,Mask R-CNN 等改进框架,但基本结构变化 ...
毕设日志2019.4.2——tf版本的faster rcnn的代码运行
今天参考https://blog.csdn.net/sinat_33486980/article/details/81045315 一文进行tf版本的faster rcnn的demo运行未果,原因是到 ...
深度学习论文阅读目标检测篇（三）：Faster R-CNN《 Towards Real-Time Object Detection with Region Proposal Networks》
深度学习论文阅读目标检测篇(三):Faster R-CNN< Towards Real-Time Object Detection with Region Proposal Networks&g ...
faster rcnn源码解读（三）train_faster_rcnn_alt_opt.py
转载自:faster rcnn源码解读(三)train_faster_rcnn_alt_opt.py - 野孩子的专栏 - 博客频道 - CSDN.NET http://blog.csdn.net/u ...
R-CNN目标检测第三弹（Faster R-CNN）
北京站 | NVIDIA DLI深度学习培训 2018年1月26日 NVIDIA 深度学习学院带你快速进入火热的DL领域阅读全文正文共5152 ...

tf-faster-rcnn指南（三）——训练模型

目录

tf-faster-rcnn指南（三）——训练模型相关推荐

最新文章

热门文章