R-CNN系列(一)

    这一部分是针对R-CNN的大致流程的概括,更多详细的说明会在后面介绍。

一、预训练

    在预训练的过程中,采用的数据集是ILSCRC样本集。ILSCRC 只有不同类别标签,没有物体对应边框的标注。预训练过程采用的是Alexnet,学习率设为0.001,输入图片大小为227*227,输出为1000类,将训练好的网络参数(除了分类层外)保存。

二、微调

    1、选择搜索

通过选择搜索策略,找出2000个候选区域

    2、给定label

将这2000个候选区域与Ground Truth比较,若IOU>0.5,则标记为该类

    3Wrap

    将候选区域做wrap操作,使得变换后的图像大小为227*227

    4、微调网络

    仍然采用Alexnet,将一中预训练好的参数直接使用,但最后的分类器改为(N+1)。分类器的初始参数随机给定,将网络训练好,保存参数。

三、SVM

    1、给定label

2000个候选区域与Ground Truth比较,若IOU<03,则标记为负样本

    2SVM 分类

将微调网络的第5个池化层后提取的特征作为SVM的输入。由于SVM是一个二分类器,故要训练NSVM

   3、非极大值抑制

通过SVM,获得2000*20维的矩阵,即每个bounding box属于各个类的概率。鉴于一个物体可能被多个bounding box包围,故采用非极大值抑制操作。

   4、边框回归

    通过与真实标注的边框对比,对极大值抑制后的结果进行边框回归


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值