Faster-RCNN 是 R-CNN 系列中的第三个版本,后续版本包括 Fast-RCNN 和原始 RCNN。由于前两个版本存在一些局限性,如 R-CNN 不能实现端到端的训练,Fast R-CNN 使用 Selective Search 方法效率较低等问题,研究者在此基础上开发了 Faster-RCNN。
Faster-RCNN 将特征提取、区域提议、边界框回归及分类任务整合到单一网络中,从而提升了检测速度,并克服了早期版本的一些缺点。
实现 Faster R-CNN 可以借助 OpenCV 和 MobileNet SSD,但需要大量代码支持。因此,我们将重点放在如何使用 OpenCV 和 Faster R-CNN 的预训练模型来实现图像和视频的目标检测。
Faster R-CNN 包含两种模型:faster_rcnn_inception_v2
和 faster_rcnn_resnet50
。前者体积较小,速度更快。本文将使用 faster_rcnn_inception_v2
模型来搭建目标检测系统。
导入模型和参数
定义模型识别类型
建立神经网络
计算 Blob 并进行预测
筛选预测结果
对于如何将此代码应用于视频检测以及如何通过多进程技术加速神经网络推理,请参阅之前的教程。
希望这些内容能帮助你更好地理解和应用 Faster-RCNN。