MonsterMMORPG commited on
Commit
33d48b5
1 Parent(s): 13f70bf

Upload setup_dense.py

Browse files
Files changed (1) hide show
  1. setup_dense.py +36 -0
setup_dense.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import re
2
+ from pathlib import Path
3
+ from setuptools import find_packages, setup
4
+
5
+ try:
6
+ import torch # noqa: F401
7
+ except ImportError as e:
8
+ raise Exception(
9
+ """
10
+ You must install PyTorch prior to installing DensePose:
11
+ pip install torch
12
+
13
+ For more information:
14
+ https://pytorch.org/get-started/locally/
15
+ """
16
+ ) from e
17
+
18
+
19
+ def get_detectron2_current_version():
20
+ return "0.6"
21
+
22
+
23
+ setup(
24
+ name="detectron2-densepose",
25
+ author="FAIR",
26
+ version=get_detectron2_current_version(),
27
+ url="https://github.com/facebookresearch/detectron2/tree/main/projects/DensePose",
28
+ packages=find_packages(),
29
+ python_requires=">=3.7",
30
+ install_requires=[
31
+ "av>=8.0.3",
32
+ "detectron2@git+https://github.com/facebookresearch/detectron2.git",
33
+ "opencv-python-headless>=4.5.3.56",
34
+ "scipy>=1.5.4",
35
+ ],
36
+ )