TensorFlow的Faster RCNN

45

有人在TensorFlow版本上实现了FRCNN吗?我找到了一些相关的repo,如下所示:

  1. 实现ROI池化层
  2. 基于py-faster-rcnn repo实现快速RCNN

但是对于1:假设ROI池化层可用(我还没有尝试),还需要实现以下内容:

  • ROI数据层,例如roidb
  • 线性回归,例如SmoothL1Loss
  • ROI池化层后处理用于端到端训练,应将ROI池化层的结果转换为CNN分类器的输入。

对于2:嗯...,它似乎基于py-faster-rcnn,后者又基于Caffe来准备预处理(例如roidb)并将数据馈送到Tensorflow中进行模型训练,这似乎很奇怪,所以我可能没有尝试过。

所以我想知道的是,Tensorflow将来是否会支持Faster RCNN?如果不支持,我是否有任何误解,如上所述?或者是否有任何存储库或个人支持它?


SmoothL1Loss应该相对容易使用实际的tf进行ROI池化实现,但我不确定... - jeandut
我正在研究与你的问题类似的目标。我发现在张量中表示动态边界框很困难。这可能是你提到的方法2使用caffe预处理数据的原因。我正在尝试找出是否有其他方法可以在TensorFlow中实现。 - Da Tong
4
这个实现怎么样?(链接为 https://github.com/smallcorgi/Faster-RCNN_TF ) - Shai
1个回答

16

Tensorflow刚刚发布了官方的目标检测API,在这里可以找到,可与其各种slim 模型一起使用。

该API包含了用于目标检测的各种流程的实现,包括流行的Faster RCNN,以及它们的预训练模型。


8
给所有的版主/审核员:请不要将这个回答作为“仅包含链接”的内容删除:问题的性质允许这样的回答,而且这个特定的回答是可以的。 - Shai
2
链接似乎已经失效。 - Gabriel Garrett
2
我已经修复了它们。 - gdelab

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接