forked from geopavlakos/hamer
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
37 lines (36 loc) · 882 Bytes
/
setup.py
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
from setuptools import setup, find_packages
print('Found packages:', find_packages())
setup(
description='HaMeR as a package',
name='hamer',
packages=find_packages(),
install_requires=[
'gdown',
'numpy',
'opencv-python',
'pyrender',
'pytorch-lightning',
'scikit-image',
'smplx==0.1.28',
'torch',
'torchvision',
'yacs',
'detectron2 @ git+https://github.com/facebookresearch/detectron2',
'chumpy @ git+https://github.com/mattloper/chumpy',
'mmcv==1.3.9',
'timm',
'einops',
'xtcocotools',
'pandas',
],
extras_require={
'all': [
'hydra-core',
'hydra-submitit-launcher',
'hydra-colorlog',
'pyrootutils',
'rich',
'webdataset',
],
},
)