diff --git a/setup.py b/setup.py new file mode 100644 index 000000000..f5b79a444 --- /dev/null +++ b/setup.py @@ -0,0 +1,43 @@ +import os +from codecs import open as copen +from setuptools import setup, find_packages + + +# Get the long description from README.md +with copen(os.path.join(os.path.abspath(os.path.dirname(__file__)), 'README.md'), mode='r', encoding='utf-8') as f: + long_description = f.read() + + +# Version +__version__ = 'v0.2' + + +setup( + name='detr', + version=__version__, + packages=find_packages(), + description="DETR: End-to-End Object Detection with Transformers", + long_description=long_description, + long_description_content_type="text/markdown", + url="https://github.com/facebookresearch/detr", + author="Facebook Research", + license='Apache License 2.0', + classifiers=[ + 'License :: OSI Approved :: Apache Software License', + 'Programming Language :: Python :: 3' + ], + # Add here the package's dependencies + install_requires=[ + 'numpy', + 'cython', + 'pycocotools', + 'submitit', + 'torch>=1.5.0', + 'torchvision>=0.6.0', + 'panopticapi @ git+https://github.com/cocodataset/panopticapi.git#egg=panopticapi', + 'scipy', + 'onnx', + 'onnxruntime', + ], + test_deps=[], +)