更快的RCNN用于TensorFlow

时间:2016-07-13 04:23:25

标签: tensorflow detection caffe pycaffe

有没有人为TensorFlow版本实现FRCNN? 我找到了一些相关的回购如下:

  1. Implement roi pool layer
  2. Implement fast RCNN based on py-faster-rcnn repo
  3. 但是1:假设roi池层工作(我没有尝试过),并且需要实现以下内容:

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

    对于2:em ....,似乎基于py-faster-rcnn基于Caffe准备预处理(例如roidb)并将数据输入Tensorflow来训练模型,这看起来很奇怪,所以我可能没想过。

    所以我想知道的是,Tensorflow support Faster RCNN in the future会是什么?如果没有,我是否有任何误解上面提到的?或者有任何回购或有人支持吗?

1 个答案:

答案 0 :(得分:12)

Tensorflow刚刚发布了一个官方的对象检测API here,例如可以使用它们的各种纤薄models

此API包含各种用于对象检测的管道的实现,包括流行的更快的RCNN,以及它们的预训练模型。