Faster-RCNN框架中alt四步训练过程

本文深入解析Faster R-CNN中的四步训练法,详细介绍了如何通过阶段性的训练优化共享卷积层参数,从而提高目标检测网络的整体性能。

对训练过程的个人理解

       Faster R-CNN中其实是训练两个网络:RPN和Fast Rcnn,两个网络如果分开单独训练,都会改变共享的卷积层的参数,所以这里需要一种技巧可以对两个网络共享卷积层,而不是分开单独训练。论文中采用一种4步训练法通过选择网络来优化参数。在代码的py-faster-rcnn/models/pascal_voc/ZF/faster_rcnn_alt_opt/有四个pt文件:stage1_rpn_train.pt,stage1_fast_rcnn_train.pt,stage2_rpn_train.pt,stage2_fast_rcnn_train.pt对应着这四步训练过程。

    1.stage1_rpn_train.pt

      单独训练RPN 网络,训练的模型用ImageNet的模型来初始化,采用end to end的方式来调整参数。

    2.stage1_fast_rcnn_train.pt

      单独训练检测网络Fast Rcnn,训练用的proposals来自第一步的RPN net,模型初始化采用ImageNet模型。

    3.stage2_rpn_train.pt

     用第二步Fast Rcnn的参数来初始化RPN模型,但是训练的时候固定卷积层,只调整属于RPN的参数。

    4.stage2_fast_rcnn_train.pt

     保持共享的卷积层固定,用第三步调整后的RPN输出的proposals作为输入,微调Fast Rcnn剩下的参数。



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值