32 lines
708 B
Python
32 lines
708 B
Python
from setuptools import setup, find_packages
|
|
|
|
print('Found packages:', find_packages())
|
|
setup(
|
|
description='HMR2 as a package',
|
|
name='hmr2',
|
|
packages=find_packages(),
|
|
install_requires=[
|
|
'gdown',
|
|
'numpy',
|
|
'torch',
|
|
'torchvision',
|
|
'pytorch-lightning',
|
|
'smplx==0.1.28',
|
|
'pyrender',
|
|
'opencv-python',
|
|
'yacs',
|
|
'scikit-image',
|
|
'einops',
|
|
'timm',
|
|
'webdataset',
|
|
'dill',
|
|
'pandas',
|
|
'chumpy @ git+https://github.com/mattloper/chumpy',
|
|
],
|
|
extras_require={
|
|
'all': [
|
|
'detectron2 @ git+https://github.com/facebookresearch/detectron2',
|
|
],
|
|
},
|
|
)
|