38 lines
882 B
Python
38 lines
882 B
Python
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',
|
|
],
|
|
},
|
|
)
|