如何在更快的RCNN中实现RPN用于目标检测?



我正在尝试实现更快的RCNN从图像中识别飞机。我坚持实施区域提案网络(RPN)。我如何实现RPN并训练他们使用python脚本生成边界框提案?

有很多现成的各种神经网络实现,包括Faster RCNN。考虑使用深度学习框架,如Pytorch或Keras。例如,查看这个Pytorch教程微调Mask R-CNN模型。

更快的RCNN是一个两阶段的目标检测模型。其中第一阶段是区域提议网络(RPN),第二阶段是分类器。对于你的任务,如果你不需要,你可以忽略第二部分。

有些实现:

Pytorch中更快的RCNN

Keras中更快的RCNN

最新更新