xfu314's picture
Add phantom project with submodules and dependencies
96da58e
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',
],
},
)