Faster RCNN AnchorTargetLayer ProposalLayer ProposalTargetLayer详细对比

Faster RCNN中三个比较重要的层,其作用位置、功能详细对比如下。三种层的出现顺序依次为AnchorTargetLayer、ProposalLayer、ProposalTargetLayer:

 Alt traingingEnd2End trainging输入框个数输出框个数
AnchorTargetLayertraintrain6000个anchor256个anchor与其对应的groundtruth
ProposalLayertrain/testtrain/test

训练:20000

测试:6000

训练:2000

测试:300

ProposalTargetLayer

(端到端训练中这部分功能RoIDataLayer实现)

/train由ProposalLayer输入的2000个proposal128个proposal与其对应的groundtruth

通过上表可以发现,AnchorTargetLayer与ProposalTargetLayer极其相似,都是只用于训练阶段,输出都是n个“框”与其对应的groundtruth,输入稍有不同,AnchorTargetLayer的输入是原始的anchor,而ProposalTargetLayer的输入是经过回归后生成的proposal。

现对两者通过下表进行详细对比:

 输入框个数输出框个数正负样本比例正负样本阈值
AnchorTargetLayer6000个anchor2561:1(正:负)

正样本:>=0.7

负样本:<0.3

与groundtruth

ProposalTargetLayer2000个proposal1281:3(正:负)

正样本:>=0.5

负样本:<0.5 && >=0.1

与groundtruth

这个还要再额外解释一点:ProposalLayer产生的RoI都是正样本,与groundtruth算了一次IoU后,根据阈值又分成了正负样本,然后送入Fast RCNN继续训练。AnchorTargetLayer、ProposalTargetLayer在测试阶段干脆就没有了。ProposalLayer直接产生300个RoI,送入Fast RCNN进行预测。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值