roi-pooling

  • Owner: deepsense-ai/roi-pooling
  • Platform:
  • License::
  • Category::
  • Topic:
  • Like:
    0
      Compare:

Github stars Tracking Chart

RoI pooling in TensorFlow

This repo contains the implementation of Region of Interest pooling as a custom TensorFlow operation. The CUDA code responsible for the computations was largely taken from the original Caffe implementation by Ross Girshick.

For more information about RoI pooling you can check out Region of interest pooling explained at our deepsense.io blog.

Region of Interest Pooling animation

Requirements

To compile and use roi_pooling layer you need to have:

Only official TensorFlow releases are currently supported. If you're using a custom built TensorFlow compiled with a different GCC version (e.g. 5.X) you may need to modify the makefile to enable the new ABI version.

Install

Since it uses compilation

$ git clone git@github.com:deepsense-io/roi-pooling.git
$ cd roi-pooling
$ python setup.py install

Right now we provide only GPU implementation (no CPU at this time).

Usage

After successful installation you can use the operation like this:

from roi_pooling.roi_pooling_ops import roi_pooling

# here obtain feature map and regions of interest
rpooling = roi_pooling(feature_map, rois, 7, 7)
# continue the model

Working example in Jupyter Notebook: examples/roi_pooling_minimal_example.ipynb

Main metrics

Overview
Name With Ownerdeepsense-ai/roi-pooling
Primary LanguageC++
Program languagePython (Language Count: 3)
Platform
License:
所有者活动
Created At2016-12-30 09:56:35
Pushed At2017-04-19 11:50:41
Last Commit At2017-04-07 15:06:56
Release Count0
用户参与
Stargazers Count460
Watchers Count11
Fork Count124
Commits Count108
Has Issues Enabled
Issues Count25
Issue Open Count17
Pull Requests Count3
Pull Requests Open Count1
Pull Requests Close Count0
项目设置
Has Wiki Enabled
Is Archived
Is Fork
Is Locked
Is Mirror
Is Private