【目标检测】Fast RCNN算法

Fast RCNN是RCNN的优化版,通过ROI pooling实现端到端训练,预训练的CNN网络加快了训练速度,多任务损失函数结合分类和边界框回归。训练时使用Selective Search提取候选框,测试时同样提取提案并进行NMS筛选,以获得最终的目标检测结果。
摘要由CSDN通过智能技术生成

论文地址:https://www.cv-foundation.org/openaccess/content_iccv_2015/papers/Girshick_Fast_R-CNN_ICCV_2015_paper.pdf

代码地址:https://github.com/rbgirshick/fast-rcnn

 Fast RCNN是在RCNN的基础上进行的改进,整体的算法思路是一样的,改进之处如以下几点:

  • 引入ROI pooling层,第一避免resize特征图到224*224,由于ROI pooling是可导的,所以实现了端到端的学习;
  • CNN网络提前做了预训练,初始化权重是进过ImageNet数据集的训练得到,大大提高网络效率;
  • 采用了多任务损失函数,一个任务是用softmax进行分类,另一个任务是边界框回归;
  • 利用截断的奇异值分解加速网络。

Fast RCNN的总体框架图:

便于理解,借鉴了一个其他人的图:

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值