py-faster-rcnn流程(4)——训练FastRCNN网络一阶段
来源:互联网 发布:冒泡法排序c语言解释 编辑:程序博客网 时间:2024/05/20 18:42
使用预训练的ImageNet模型和上一步产生的ROI初始化Fastrcnn网络
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
这一步,用上一步生成的proposal,以及imagenet模型M0来训练fast-rcnn模型M2。
关注train_fast_rcnn
同样地,会设置参数,这里注意cfg.TRAIN.PROPOSAL_METHOD = ‘rpn’ 不同于前面,后面调用的将是rpn_roidb。cfg.TRAIN.IMS_PER_BATCH = 2,每个mini-batch包含两张图片,以及它们proposal的roi区域。且在这一步是有rpn_file的(后面和rpn_roidb函数使用有关)。其他的和前面差不多。提一下,这里在train_net的时候,会调用add_bbox_regression_targets位于roidb中,主要是添加bbox回归目标,即添加roidb的‘bbox_targets’属性,同时根据cfg中的参数设定,求取bbox_targets的mean和std,因为需要训练class-specific regressors在这里就会涉及到bbox_overlaps函数,放在util.bbox中。
要注意的是在这一步get_roidb时,如前所说,使用的是rpn_roidb,会调用imdb. create_roidb_from_box_list该方法功能是从box_list中读取每张图的boxes,而这个box_list就是从上一步保存的proposal文件中读取出来的,然后做一定的处理,详细见代码,重点是在最后会返回roidb,rpn_roidb中的gt_overlaps是rpn_file中的box与gt_roidb中box的gt_overlaps等计算IoU等处理后得到的,而不像gt_roidb()方法生成的gt_roidb中的gt_overlaps全部为1.0。同时使用了imdb.merge_roidb,把rpn_roidb和gt_roidb归并为一个roidb。
- py-faster-rcnn流程(4)——训练FastRCNN网络一阶段
- py-faster-rcnn流程(2)——训练RPN网络一阶段
- py-faster-rcnn流程(6)——训练Fastrcnn网络二阶段
- py-faster-rcnn流程(5)——训练RPN网络二阶段
- py-faster-rcnn流程(3)——根据RPN网络的输出获取ROI
- py-faster-rcnn流程(1)——准备阶段
- 训练py-faster-rcnn
- 训练py-faster-rcnn(caffe)
- py-faster-rcnn训练教程
- Faster-RCNN Tensorflow版本源码解析(一):网络训练部分train_net.py
- py-faster-rcnn详解(4)——anchor_target_layer.py接口说明
- 使用faster rcnn训练自己的数据(py-faster-rcnn )
- py-faster-rcnn训练自己的数据
- py-faster-rcnn 训练常见错误
- py faster rcnn训练 VGG_CNN_M_1024记录
- py-faster-rcnn end2end训练记录
- Tensorflow框架下Faster-RCNN实践(二)——用自己制作的数据训练Faster-RCNN网络(附代码)
- Tensorflow框架下Faster-RCNN实践(二)——用自己制作的数据训练Faster-RCNN网络(附代码)
- 网站静态化处理—前后端分离—下(8)
- hdu1023 Train Problem II(卡特兰数)
- python 关于list.append()容易犯的错误
- 淘淘商城系列——使用SolrJ管理Solr集群
- 进程调度之--进程的组织
- py-faster-rcnn流程(4)——训练FastRCNN网络一阶段
- Hyperledger-Fabric技术文档1--Fabric初体验
- 天气开发2——第二行代码(酷欧天气)
- 网站静态化处理—满足静态化的前后端分离(9)
- pxe
- Java 多线程之内置锁与显示锁
- 秒懂Java回调函数
- C/C++_log2000_2017春季算法实验1_3
- Linux运维笔记-文档总结-Samba文件共享