仅使用Tensorflow进行区域投标网络培训以进行对象检测

Question

我喜欢使用RPN network从faster rcnn训练tensorflow。 在咖啡方面，我只训练了RPN。

在tensorflow中，this tutorial显示训练自定义网络。然后the other tutorial在这里讨论了如何使用modelzoo中的模型进行训练。

那么对于我来说，只训练RPN零件，哪种方法适合我？ 假设我遵循定义CNN网络的第一种方法，如何为pretrained model加载transfer learning。

Answer 1

要仅使用张量流对象检测api训练RPN网络，只需更改配置文件，因为该API支持here中所述的仅RPN选项。要获得仅RPN的模型，只需在模型配置area中添加选项'number_of_stages : 1'（类似于'num_of_classes : 90'）。 （您也可以从配置文件中删除一些第二阶段参数，因为如果number_of_stages设置为1，则它们将无用。）

关于转移学习的第二个问题，就像训练faster rcnn网络一样，只需确保from_detection_checkpoint是正确的，并且finetune_checkpoint_path在{ {3}}。

在训练过程中，您可以使用张量板检查状态，执行评估后，您可以在“图像”选项卡中看到区域建议在图像上可视化，并且标签均为“对象+置信度得分”。