Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/env python
- import glob
- import os
- import shutil
- from os import path
- from setuptools import find_packages, setup
- def getD2Version():
- init_py_path = path.join(path.abspath(path.dirname(__file__)), "detectron2", "__init__.py")
- init_py = open(init_py_path, "r").readlines()
- version_line = [l.strip() for l in init_py if l.startswith("__version__")][0]
- version = version_line.split("=")[-1].strip().strip("'\"")
- return version
- setup(
- name="densepose",
- version=0.1,
- author="FAIR",
- url="https://github.com/facebookresearch/detectron2",
- description="Detectron2 is FAIR's next-generation research "
- "platform for object detection and segmentation.",
- python_requires=">=3.6",
- install_requires=[
- # These dependencies are not pure-python.
- # In general, avoid adding more dependencies like them because they are not
- # guaranteed to be installable by `pip install` on all platforms.
- # To tell if a package is pure-python, go to https://pypi.org/project/{name}/#files
- "Pillow>=7.1", # or use pillow-simd for better performance
- "matplotlib", # TODO move it to optional after we add opencv visualization
- "pycocotools>=2.0.2", # corresponds to https://github.com/ppwwyyxx/cocoapi
- "av",
- # Do not add opencv here. Just like pytorch, user should install
- # opencv themselves, preferrably by OS's package manager, or by
- # choosing the proper pypi package name at https://github.com/skvark/opencv-python
- # The following are pure-python dependencies that should be easily installable
- "termcolor>=1.1",
- "yacs>=0.1.6",
- "tabulate",
- "cloudpickle",
- "tqdm>4.29.0",
- "tensorboard",
- "tqdm",
- # Lock version of fvcore/iopath because they may have breaking changes
- # NOTE: when updating fvcore/iopath version, make sure fvcore depends
- # on compatible version of iopath.
- "fvcore>=0.1.5,<0.1.6", # required like this to make it pip installable
- "iopath>=0.1.7,<0.1.10",
- "future", # used by caffe2
- "pydot", # used to save caffe2 SVGs
- "dataclasses; python_version<'3.7'",
- "omegaconf>=2.1",
- "hydra-core>=1.1",
- "black==21.4b2",
- # If a new dependency is required at import time (in addition to runtime), it
- # probably needs to exist in docs/requirements.txt, or as a mock in docs/conf.py
- ]
- )
Add Comment
Please, Sign In to add comment