Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,247 Bytes
61c2d32 |
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 43 |
import re
from pathlib import Path
from setuptools import find_packages, setup
try:
import torch # noqa: F401
except ImportError as e:
raise Exception(
"""
You must install PyTorch prior to installing DensePose:
pip install torch
For more information:
https://pytorch.org/get-started/locally/
"""
) from e
def get_detectron2_current_version():
"""Version is not available for import through Python since it is
above the top level of the package. Instead, we parse it from the
file with a regex."""
# Get version info from detectron2 __init__.py
version_source = (Path(__file__).parents[2] / "detectron2" / "__init__.py").read_text()
version_number = re.findall(r'__version__ = "([0-9\.]+)"', version_source)[0]
return version_number
setup(
name="detectron2-densepose",
author="FAIR",
version=get_detectron2_current_version(),
url="https://github.com/facebookresearch/detectron2/tree/main/projects/DensePose",
packages=find_packages(),
python_requires=">=3.7",
install_requires=[
"av>=8.0.3",
# "detectron2@git+https://github.com/facebookresearch/detectron2.git",
"opencv-python-headless>=4.5.3.56",
"scipy>=1.5.4",
],
)
|