更快的RCNN用于TensorFlow

问题描述 投票:42回答:1

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

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

但是对于1:假设roi池层工作(我还没有尝试过),并且需要实现如下内容:

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

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

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

tensorflow detection caffe pycaffe
1个回答
12
投票

Tensorflow刚刚发布了官方的对象检测API here,可以用于各种纤薄的models

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

© www.soinside.com 2019 - 2024. All rights reserved.