setup.py

#!/usr/bin/env python
import re

import setuptools


def get_info():
    with open("README.md", "r") as fh:
        long_description = fh.read()
        version = re.search(
            r'^__version__\s*=\s*"(.*)"',
            open("experimenting/__init__.py").read(),
            re.MULTILINE,
        ).group(1)
    return long_description, version


long_description, version = get_info()

setuptools.setup(
    name='lifting_events_to_3d_hpe',
    version=version,
    description='Code for 2021 CVPRw "Lifting Monocular Events to 3D Human Poses"',
    long_description=long_description,
    author='Gianluca Scarpellini',
    author_email='gianluca.scarpellini@iit.it',
    url='https://github.com/iit-pavis/lifting_events_to_3d_hpe',
    install_requires=[
        'albumentations',
        'opencv-python==4.2.0.34',
        'h5py',
        'scikit-image',
        'scikit-learn',
        'scikit-video',
        'scipy',
        'torch>1.4',
        'kornia',
        'hydra-core==1.0.0rc1',
        'omegaconf',
        'pytorch-lightning==1.1.6',
        'torchvision',
        'tqdm',
        'segmentation_models_pytorch',
    ],
    packages=setuptools.find_packages(exclude=["tests", "tests/*"]),
    test_suite="tests",
    long_description_content_type="text/markdown",
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: GPL License",
        "Operating System :: Linux",
    ],
)