2016-07-13 31 views
40

有没有人为TensorFlow版本实现FRCNN? 我发现了一些相关的回购协议如下:TensorFlow更快的RCNN

  1. Implement roi pool layer
  2. Implement fast RCNN based on py-faster-rcnn repo

,但1:假设投资回报率汇集层工作(我没试过),有什么需要实施如下:

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

对于2:em ....似乎基于py-faster-rcnn,它基于Caffe准备预处理(例如roidb)并将数据馈入Tensorflow以训练模型,它似乎很奇怪,所以我可能没有尝试过。我想知道的是,Tensorflow support Faster RCNN in the future?。如果没有,我是否有任何上面提到的错误理解?或有任何回购或某人的支持?

+0

SmoothL1Loss应该相对容易实现使用ROI池的实际tf不知道... – jean

+0

我正在处理您的问题的类似目标。我发现很难用张量表示动态bbox。这也许就是你提到的方法2使用caffe来预处理数据的原因。我试图找出在TensorFlow中是否有其他方法来实现这一点。 –

+4

[此实现]如何(https://github.com/smallcorgi/Faster-RCNN_TF)? – Shai

回答

10

Tensorflow刚刚发布了一个官方的对象检测API here,它可以用于他们的各种苗条models

该API包含各种管道对象检测的实现,包括受欢迎的快速RCNN以及其预先训练的模型。

+6

所有mod/reviewers:请做**不**删除这个答案为“仅链接”:问题的性质允许这样的答案,这个特定的答案是好的。 – Shai

+1

链接似乎不存在 – zavtra

+2

我修复了它们 – gdelab