R-CNN

本文相关

  • Paper Summary:https://github.com/FDU-VTS/CVPaper
  • Code:https://github.com/FDU-VTS/CVCode

原文链接

  • https://arxiv.org/abs/1311.2524

定义

  • region proposal(region of interest): 候选区域
  • ground truth: 样本区域
  • IoU: 计算重叠面积 = (A∩B) / (A∪B)

算法流程

CNN训练

  • 数据集准备
    • 选取训练集中的ground truth,并使用selective search计算出region proposals
    • 计算ground truthroi之间的IoU,选取其中值大于0.5的rois,然后选取25%计入正样本中
    • 第二步中IoU小于0.5中选取正样本3倍的数量记为background
  • 训练网络
    • alex_net/VGG-16
  • 损失函数
    • cross entropy

SVM训练

  • 数据集准备
    • 按照CNN训练中的,不过这次挑选IoU大于0.5的proposals, 和ground truth一起组成正样本集合
    • IoU小于0.3的归为负样本集合
  • 训练m个类别的SVM

bounding box regression

  • 坐标和尺度变换:
  • 损失函数
  • 正确值

你可能感兴趣的:(R-CNN)