File size: 1,129 Bytes
46a3752 a4d0487 46a3752 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
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,
},
) |