detection_metrics / setup.py
AlekseyKorshuk's picture
Update setup.py
a4d0487 verified
from detection_metrics import __version__
import subprocess
from setuptools import setup
from setuptools.command.develop import develop
from setuptools.command.egg_info import egg_info
from setuptools.command.install import install
def custom_command():
subprocess.call(["pip", "install", "numpy", "cython"])
subprocess.call(["pip", "install", "-r", "requirements.txt", "--user"])
class CustomInstallCommand(install):
def run(self):
install.run(self)
custom_command()
class CustomDevelopCommand(develop):
def run(self):
develop.run(self)
custom_command()
class CustomEggInfoCommand(egg_info):
def run(self):
egg_info.run(self)
custom_command()
setup(
name="detection_metrics",
description="COCO Metrics for Object Detection and Instance Segmentation",
version=__version__,
zip_safe=True,
# packages=["detection_metrics", "detection_metrics.pycocotools"],
include_package_data=True,
cmdclass={
"install": CustomInstallCommand,
"develop": CustomDevelopCommand,
"egg_info": CustomEggInfoCommand,
},
)