Spaces:
Running
Running
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, | |
}, | |
) |