如何在 Faster RCNN 中实现 RPN 进行目标检测？

【问题标题】：How to implement RPN in Faster RCNN for object detection?如何在 Faster RCNN 中实现 RPN 进行目标检测？
【发布时间】：2021-10-14 12:48:07
【问题描述】：

我正在尝试使用 Faster RCNN 从图像中识别飞机。我坚持实施区域提案网络（RPN）。如何实现 RPN 并训练他们使用 python 脚本生成边界框建议？

【问题讨论】：

您确定要重新实现它，而不是像这样使用torchvision 模型（或Keras 等效模型）：pytorch.org/tutorials/intermediate/…

【解决方案1】：

包括 Faster RCNN 在内的各种神经网络都有大量现成的实现。考虑使用 DL 框架，例如 Pytorch 或 Keras。例如，请参阅this Pytorch tutorial on fine-tuning the Mask R-CNN model。

Faster RCNN 是一个两阶段的目标检测模型。其中第一阶段是RPN（Region Proposal Network），第二阶段是分类器。对于您的任务，如果您不需要第二部分，您可以忽略它。

一些实现：

【讨论】：