TRACE/setup.py at master · Barath19/TRACE · GitHub

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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import setuptools
from distutils.core import setup, Extension
from Cython.Build import cythonize
import numpy

with open("README.md", "r", encoding="utf-8") as fh:
    long_description = fh.read()

requireds = ["opencv-python","torch",
        'setuptools>=18.0.0',
        'cython',
        'numpy>=1.21.0',
        'typing-extensions>=4.1',
        'scipy',
        'lap']

setuptools.setup(
    name='simple_romp',
    version='1.1.3',
    author="Yu Sun",
    author_email="yusunhit@gmail.com",
    setup_requires=[
        # Setuptools 18.0 properly handles Cython extensions.
        'setuptools>=18.0.0',
        'cython',
        'numpy>=1.21.0',
        'typing-extensions>=4.1',
        'scipy',
        'lap'],
    install_requires=requireds,
    description="ROMP [ICCV21], BEV [CVPR22], TRACE [CVPR23]",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/Arthur151/ROMP",
    packages=[
        'romp',
        'vis_human',
        'vis_human.sim3drender',
        'vis_human.sim3drender.lib',
        'bev',
        'trace2',
        'trace2.tracker',
        'trace2.models',
        'trace2.models.raft',
        'trace2.models.raft.utils',
        'trace2.models.deform_conv',
        'trace2.models.deform_conv.functions',
        'trace2.results_parser',
        'trace2.evaluation',
        'trace2.evaluation.dynacam_evaluation',
        'trace2.evaluation.TrackEval',
        'trace2.evaluation.TrackEval.trackeval',
        'trace2.evaluation.TrackEval.trackeval.metrics',
        'trace2.evaluation.TrackEval.trackeval.datasets',
        'trace2.evaluation.TrackEval.trackeval.baselines',
        'trace2.utils',
        'tracker'],
    ext_modules=cythonize([Extension("Sim3DR_Cython",
                           sources=["vis_human/sim3drender/lib/rasterize.pyx",
                                    "vis_human/sim3drender/lib/rasterize_kernel.cpp"],
                           language='c++',
                           include_dirs=[numpy.get_include()],
                           extra_compile_args=["-std=c++11"])]),
    include_package_data=True,
    classifiers=[
         "Programming Language :: Python :: 3",
         "License :: Other/Proprietary License",
         "Operating System :: OS Independent",
    ],
    project_urls={
        "Bug Tracker": "https://github.com/Arthur151/ROMP/issues",
    },
    entry_points={
        "console_scripts": [
            "romp=romp.main:main",
            "bev=bev.main:main",
            "trace2=trace2.main:main",
            "romp.prepare_smpl=romp.pack_smpl_info:main",
            "bev.prepare_smil=bev.pack_smil_info:main",
        ],
    },
)