jfngsea

google_colab_densepose_d2

Jul 24th, 2021 (edited)
1,317
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.47 KB | None | 0 0
  1. #!/usr/bin/env python
  2. import glob
  3. import os
  4. import shutil
  5. from os import path
  6. from setuptools import find_packages, setup
  7.  
  8. def getD2Version():
  9.     init_py_path = path.join(path.abspath(path.dirname(__file__)), "detectron2", "__init__.py")
  10.     init_py = open(init_py_path, "r").readlines()
  11.     version_line = [l.strip() for l in init_py if l.startswith("__version__")][0]
  12.     version = version_line.split("=")[-1].strip().strip("'\"")
  13.  
  14.     return version
  15.  
  16. setup(
  17.     name="densepose",
  18.     version=0.1,
  19.     author="FAIR",
  20.     url="https://github.com/facebookresearch/detectron2",
  21.     description="Detectron2 is FAIR's next-generation research "
  22.     "platform for object detection and segmentation.",
  23.     python_requires=">=3.6",
  24.     install_requires=[
  25.         # These dependencies are not pure-python.
  26.         # In general, avoid adding more dependencies like them because they are not
  27.         # guaranteed to be installable by `pip install` on all platforms.
  28.         # To tell if a package is pure-python, go to https://pypi.org/project/{name}/#files
  29.         "Pillow>=7.1",  # or use pillow-simd for better performance
  30.         "matplotlib",  # TODO move it to optional after we add opencv visualization
  31.         "pycocotools>=2.0.2",  # corresponds to https://github.com/ppwwyyxx/cocoapi
  32.         "av",
  33.         # Do not add opencv here. Just like pytorch, user should install
  34.         # opencv themselves, preferrably by OS's package manager, or by
  35.         # choosing the proper pypi package name at https://github.com/skvark/opencv-python
  36.         # The following are pure-python dependencies that should be easily installable
  37.         "termcolor>=1.1",
  38.         "yacs>=0.1.6",
  39.         "tabulate",
  40.         "cloudpickle",
  41.         "tqdm>4.29.0",
  42.         "tensorboard",
  43.         "tqdm",
  44.         # Lock version of fvcore/iopath because they may have breaking changes
  45.         # NOTE: when updating fvcore/iopath version, make sure fvcore depends
  46.         # on compatible version of iopath.
  47.         "fvcore>=0.1.5,<0.1.6",  # required like this to make it pip installable
  48.         "iopath>=0.1.7,<0.1.10",
  49.         "future",  # used by caffe2
  50.         "pydot",  # used to save caffe2 SVGs
  51.         "dataclasses; python_version<'3.7'",
  52.         "omegaconf>=2.1",
  53.         "hydra-core>=1.1",
  54.         "black==21.4b2",
  55.         # If a new dependency is required at import time (in addition to runtime), it
  56.         # probably needs to exist in docs/requirements.txt, or as a mock in docs/conf.py
  57.     ]
  58. )
Add Comment
Please, Sign In to add comment