基于源码的Faster R-CNN训练总体流程梳理

来源:互联网 发布:二维码生成打印软件 编辑:程序博客网 时间:2024/05/18 01:30

之前的文章中对于Faster R-CNN的总体流程进行了梳理,详情参见这里。本篇文章将要以源码中的结构为基础,对于训练的整体流程进行梳理,以便后续源码的阅读。

首先我们要明确的是,实际的Faster R-CNN中训练分为6个步骤:

  1. 第一次训练RPN网络,对应源码中的stage1_rpn_train.pt
  2. 利用步骤1训练好的RPN网络,收集proposals,对应rpn_test.pt
  3. 第一次训练Fast RCNN网络,对应stage1_fast_rcnn_train.pt
  4. 第二次训练RPN网络,对应stage2_rpn_train.pt
  5. 再次利用步骤4中训练好的RPN网络,收集proposals,对应rpn_test.pt
  6. 第二次训练Fast RCNN网络,对应stage2_fast_rcnn_train.pt

上述的迭代训练过程使用了两次循环,因为作者通过实践表明,再进行更多的循环网络的效果并不会再得到提升了。

由于两次迭代的过程基本相同,因此仅仅分析第一次迭代过程。

RPN网络训练

这里写图片描述

Proposals收集

这里写图片描述

Fast RCNN网络训练

阅读全文
0 0
原创粉丝点击