Faster RCNN for TensorFlow

Question

Has anyone implement the FRCNN for TensorFlow version? I found some related repos as following:

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

but for 1: assume the roi pooling layer works (I haven't tried), and there are something need to be implemented as following:

• ROI data layer e.g. roidb.
• Linear Regression e.g. SmoothL1Loss
• ROI pool layer post-processing for end-to-end training which should convert the ROI pooling layer's results to feed into CNN for classifier.

For 2: em...., it seems based on py-faster-rcnn which based on Caffe to prepared pre-processing (e.g. roidb) and feed data into Tensorflow to train the model, it seems weird, so I may not tried it.

So what I want to know is that, will Tensorflow support Faster RCNN in the future?. If not, do I have any mis-understand which mentioned above? or has any repo or someone support that?

Answer 1

Tensorflow has just released an official Object Detection API here, that can be used for instance with their various slim models.

This API contains implementation of various Pipelines for Object Detection, including popular Faster RCNN, with their pre-trained models as well.