setup.py

#!/usr/bin/env python

from __future__ import print_function
from distutils.core import setup
from distutils.command.install import install as DistutilsInstall
import sys
import subprocess

try:
    import tensorflow
except ImportError:
    print("Please install tensorflow 0.12.0 or later")
    sys.exit()
    

class MyInstall(DistutilsInstall):
    def run(self):
        subprocess.call(['make', '-C', 'roi_pooling', 'build'])
        DistutilsInstall.run(self)

setup(name='roi-pooling',
            version='1.0',
            description='ROI pooling as a custom TensorFlow operation',
            author='deepsense.io',
            packages=['roi_pooling'],
            package_data={'roi_pooling': ['roi_pooling.so']},
            cmdclass={'install': MyInstall}
)