目标检测算法——R-CNN

简介: 目标检测算法——R-CNN

1.R-CNN简介


R-CNN(Region with CNN feature)

文章的全称为:Rich feature hierarchies for accurate object detection andsemantic segmentation,原文链接

翻译过来就是针对高准确度的目标检测与语义分割的多特征层级,通俗地来讲就是一个用来做目标检测和语义分割的神经网络。


这篇论文发布时间是 2014 年,它具有很多比较重要的意义。

  1. 在 Pascal VOC 2012 的数据集上,能够将目标检测的验证指标 mAP 提升到 53.3%,这相对于之前最好的结果提升了整整 30%.
  2. 这篇论文证明了可以讲神经网络应用在自底向上的候选区域,这样就可以进行目标分类和目标定位。
  3. 这篇论文也带来了一个观点,那就是当你缺乏大量的标注数据时,比较好的可行的手段是,进行神经网络的迁移学习,采用在其他大型数据集训练过后的神经网络,然后在小规模特定的数据集中进行 fine-tune 微调。


RCNN 算法流程可分为4个步骤

  • 一张图像生成1K~2K个 候选区域(使用Selective Search方法)
  • 对每个候选区域,使用深度网络 提取特征
  • 特征送入每一类的 SVM 分类器,判别是否属于该类
  • 使用回归器精细修正候选框位置

image.png


2.R-CNN处理过程


1)候选区域的生成

能够生成候选区域的方法有很多,比如:

  • objectness
  • selective search
  • category-independen object proposals
  • constrained parametric min-cuts(CPMC)
  • multi-scale combinatorial grouping
  • Ciresan

其中,C-RNN采用是Selective Search算法(ss算法)。

利用Selective Search算法通过图像分割的方法得到一些原始区域,然后使用一些合并策略将这些区域合并,得到一个层次化的区域结构,而这些结构就包含着可能需要的物体。

image.png


2)对每个候选区域,使用深度网络提取特征

将由第一步ss算法框选的2000个候选区域缩放到相同的尺寸227x227pixel,接着将候选区域输入事先训练好的AlexNet CNN网络获取4096维的特征得到2000×4096维矩阵。该矩阵的每一行代表了一个候选框的通过卷积网络后提取到的特征信息。

image.png


3)特征信息送入每一类的 SVM 分类器,判定类别

将2000×4096维特征与20个SVM组成的权值矩阵4096×20相乘,获得2000×20维矩阵表示每个建议框是某个目标类别的得分。

image.png

SVM的权值矩阵中的每一列代表了一类的权值;而相乘后所得到的矩阵的每一行,代表了一个候选框的20个类别的概率,2000表示一共有2000个候选框。

比如:第一行的第一个值代表了该第一个候选框是猫的概率,第二个值表示该候选框是狗的概率。而第二行的第一个值就代表了第二个候选框是猫的概率。

image.png

然后分别对上述2000×20维矩阵中每一列即每一类进行非极大值抑制剔除重叠建议框,得到该列即该类中得分最高的一些建议框。


非极大值抑制剔除重叠建议框

IoU(Intersection over Union)表示(A∩B)/(A∪B),也就是两个区域的交并比。

image.png

具体操作是:

对于每一类中,有2000个候选框,其中我们寻找其中得分最高的目标,然后计算其他目标与该目标的iou值,然后删除所有iou值大于给定阈值的目标。因为如果iou值大于给定阈值,就说明了这两者其中是可以被看作是同一个目标,所以可以选择得分高的,然后删除其他得分低的。如此反复循环,不断地减少候选框的个数。

image.png

如此操作后,希望得到的是一个最完美的边界框。

以上的操作,是对2000x20维的概率矩阵的每一列,都做刚刚所说的非极大值抑制处理。因为每一列是对于同一个类别的概率。


4)使用回归器精细修正候选框位置

通过ss算法得到的目标候选框其实位置不算是那么的准确,所以需要通过回归器修正位置。


对NMS处理后剩余的建议框进一步筛选。具体做法是,保留与真实标准的边界框有相交的,并且iou要大于某一个阈值,不满足就要将其删除掉。


接着分别用20个回归器对上述20个类别中剩余的建议框进行回归操作,最终得到每个类别的修正后的得分最高的bounding box。这是针对卷积神经网络输出的特征向量进行预测的。利用每一个边界框4096维的特征信息来进行预测的。


通过回归分类器之后,会得到四个参数,分别得到目标建议框中心的x偏移量与y偏移量,以及边界框的高度缩放因子与宽度缩放因子。一共四个值,通过这四个值来对建议框进行调整,得到一个红色的建议框。


如图,黄色框口P表示通过ss算法得到的候选建议框Region Proposal,绿色窗口G表示实际框Ground Truth,红色窗口表示Region Proposal进行回归后的预测窗口,可以用最小二乘法解决的线性回归问题。

image.png

R-CNN框架

image.png


3.R-CNN存在的问题


  1. 测试速度慢:

测试一张图片约53s(CPU)。用Selective Search算法提取候选框用时约2秒,一张图像内候选框之间存在大量重叠,提取特征操作冗余。因为得到的这些候选框会有很多重叠的部分。

  1. 训练速度慢:

过程及其繁琐。不仅要训练图像分类网络,还有训练SVM分类器,还有训练回归器,这几个模型的训练都是独立的,所以费时费力。

  1. 训练所需空间大:

对于SVM和bbox回归训练,需要从每个图像中的每个目标候选框提取特征,并写入磁盘。对于非常深的网络,如VGG16,从VOC07训练集上的5k图像上提取的特征需要数百GB的存储空间。


参考资料:

b站u主视频:https://www.bilibili.com/video/BV1af4y1m7iL

https://blog.csdn.net/briblue/article/details/820125 75

目录
相关文章
|
2天前
|
机器学习/深度学习 数据采集 算法
基于GWO灰狼优化的CNN-GRU-SAM网络时间序列回归预测算法matlab仿真
本项目基于MATLAB2022a,展示了时间序列预测算法的运行效果(无水印)。核心程序包含详细中文注释和操作视频。算法采用CNN-GRU-SAM网络,结合灰狼优化(GWO),通过卷积层提取局部特征、GRU处理长期依赖、自注意力机制捕捉全局特征,最终实现复杂非线性时间序列的高效预测。
|
4月前
|
机器学习/深度学习 算法 数据安全/隐私保护
基于MSER和HOG特征提取的SVM交通标志检测和识别算法matlab仿真
### 算法简介 1. **算法运行效果图预览**:展示算法效果,完整程序运行后无水印。 2. **算法运行软件版本**:Matlab 2017b。 3. **部分核心程序**:完整版代码包含中文注释及操作步骤视频。 4. **算法理论概述**: - **MSER**:用于检测显著区域,提取图像中稳定区域,适用于光照变化下的交通标志检测。 - **HOG特征提取**:通过计算图像小区域的梯度直方图捕捉局部纹理信息,用于物体检测。 - **SVM**:寻找最大化间隔的超平面以分类样本。 整个算法流程图见下图。
|
1月前
|
机器学习/深度学习 数据采集 算法
基于GA遗传优化的CNN-GRU-SAM网络时间序列回归预测算法matlab仿真
本项目基于MATLAB2022a实现时间序列预测,采用CNN-GRU-SAM网络结构。卷积层提取局部特征,GRU层处理长期依赖,自注意力机制捕捉全局特征。完整代码含中文注释和操作视频,运行效果无水印展示。算法通过数据归一化、种群初始化、适应度计算、个体更新等步骤优化网络参数,最终输出预测结果。适用于金融市场、气象预报等领域。
基于GA遗传优化的CNN-GRU-SAM网络时间序列回归预测算法matlab仿真
|
2月前
|
机器学习/深度学习 数据采集 算法
基于PSO粒子群优化的CNN-GRU-SAM网络时间序列回归预测算法matlab仿真
本项目展示了基于PSO优化的CNN-GRU-SAM网络在时间序列预测中的应用。算法通过卷积层、GRU层、自注意力机制层提取特征,结合粒子群优化提升预测准确性。完整程序运行效果无水印,提供Matlab2022a版本代码,含详细中文注释和操作视频。适用于金融市场、气象预报等领域,有效处理非线性数据,提高预测稳定性和效率。
|
2月前
|
机器学习/深度学习 算法 数据安全/隐私保护
基于贝叶斯优化CNN-GRU网络的数据分类识别算法matlab仿真
本项目展示了使用MATLAB2022a实现的贝叶斯优化、CNN和GRU算法优化效果。优化前后对比显著,完整代码附带中文注释及操作视频。贝叶斯优化适用于黑盒函数,CNN用于时间序列特征提取,GRU改进了RNN的长序列处理能力。
|
3月前
|
机器学习/深度学习 监控 算法
基于反光衣和检测算法的应用探索
本文探讨了利用机器学习和计算机视觉技术进行反光衣检测的方法,涵盖图像预处理、目标检测与分类、特征提取等关键技术。通过YOLOv5等模型的训练与优化,展示了实现高效反光衣识别的完整流程,旨在提升智能检测系统的性能,应用于交通安全、工地监控等领域。
|
4月前
|
机器学习/深度学习 算法 数据安全/隐私保护
基于贝叶斯优化CNN-LSTM网络的数据分类识别算法matlab仿真
本项目展示了基于贝叶斯优化(BO)的CNN-LSTM网络在数据分类中的应用。通过MATLAB 2022a实现,优化前后效果对比明显。核心代码附带中文注释和操作视频,涵盖BO、CNN、LSTM理论,特别是BO优化CNN-LSTM网络的batchsize和学习率,显著提升模型性能。
|
4月前
|
监控 算法 数据安全/隐私保护
基于三帧差算法的运动目标检测系统FPGA实现,包含testbench和MATLAB辅助验证程序
本项目展示了基于FPGA与MATLAB实现的三帧差算法运动目标检测。使用Vivado 2019.2和MATLAB 2022a开发环境,通过对比连续三帧图像的像素值变化,有效识别运动区域。项目包括完整无水印的运行效果预览、详细中文注释的代码及操作步骤视频,适合学习和研究。
|
4月前
|
JSON 算法 数据可视化
测试专项笔记(一): 通过算法能力接口返回的检测结果完成相关指标的计算(目标检测)
这篇文章是关于如何通过算法接口返回的目标检测结果来计算性能指标的笔记。它涵盖了任务描述、指标分析(包括TP、FP、FN、TN、精准率和召回率),接口处理,数据集处理,以及如何使用实用工具进行文件操作和数据可视化。文章还提供了一些Python代码示例,用于处理图像文件、转换数据格式以及计算目标检测的性能指标。
109 0
测试专项笔记(一): 通过算法能力接口返回的检测结果完成相关指标的计算(目标检测)
|
4月前
|
算法 安全
分别使用OVP-UVP和OFP-UFP算法以及AFD检测算法实现反孤岛检测simulink建模与仿真
本课题通过Simulink建模与仿真,实现OVP-UVP、OFP-UFP算法及AFD检测算法的反孤岛检测。OVP-UVP基于电压幅值变化,OFP-UFP基于频率变化,而AFD则通过注入频率偏移信号来检测孤岛效应,确保电力系统安全稳定运行。系统使用MATLAB 2013b进行建模与仿真验证。