算法背景
论文全称:Cascade R-CNN: Delving into High Quality Object Detection
论文链接:https://arxiv.org/abs/1712.00726
论文时间:2017.12.3
本文是在R-CNN算法上的拓展,针对目标检测问题,我们利用IoU(intersection over union)来定于正样本与负样本,IoU的阈值的定义是非常重要的。IoU阈值太低,eg.0.5,会产生很多带有噪声的样本。
问题
但随着阈值的升高也会导致检测表现下降:
- IoU阈值太高,会导致正样本的数量太少,从而导致过拟合(overfit)。正样本数量会呈指数下降。
- 训练与推断使用不一样的阈值会导致不匹配现象。
解决方案
本文提出了一个多阶段的目标检测模型。核心就是利用不断提高的阈值,在保证样本数不减少的情况下训练出高质量的检测器,通过级联检测网络来达到由于预测结果的目的。优化的假设的重采样确保所有的检测器都有一个相当尺寸的正样本集,从而减少过拟合问题。
算法介绍
目标检测问题可以被分为分类与定位两个复杂问题,定位问题由于近似误分类为正样本( “close” false positives)问题而变得难以解决。这些边界框“close but not correct”。
现阶段的提取候选区域的目标检测算法都是两阶段的