video-transformer/setup.py at main · Simcs/video-transformer · 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
from setuptools import setup, find_packages

setup(
    name='video-transformer',
    version='1.0.0',
    author='Simcs',
    author_email='smh3946@gmail.com',
    url='https://github.com/Simcs/video-transformer.git',
    classifiers=[
        'Programming Language :: Python :: 3.8',
        'Programming Language :: Python :: 3 :: Only'
    ],
    packages=find_packages(),
    install_requires=[
        'torch',
        'torchmetrics==0.5.1',
        'torchvision',
        'pytorch-lightning==1.3.8',
        'pytorchvideo',
        'scikit-image',
        'decord==0.6.0',
        'einops==0.3.2',
        'kornia==0.6.1',
        'matplotlib==3.4.3',
        'timm==0.4.12',
    ],
    python_requires='>=3.8',
)