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

setup(
    name="beyondfid",
    version="0.1.0",
    packages=find_packages(),
    entry_points={
        "console_scripts": [
            "beyondfid=beyondfid.run:main",
            "beyondfid-check=beyondfid.check:main",
        ],
    },
    install_requires=[
        # Core — always required
        'torch>=2.0.0',
        'torchvision>=0.15.0',
        'ml_collections>=0.1.1',
        'pandas>=1.5.0',
        'matplotlib>=3.7.0',
        'opencv-python-headless>=4.8.0',
        'scikit-learn>=1.3.0',
        'numpy>=1.24.0',
        'pillow>=10.0.0',
        'tqdm>=4.65.0',
        'h5py>=3.0.0',
    ],
    extras_require={
        # Model-specific — only needed when that feature extractor is used
        # pip install "beyondfid[transformers]"  → data2vec
        'transformers': ['transformers>=4.30.0'],
        # pip install "beyondfid[diffusers]"     → sdvae
        'diffusers': ['diffusers>=0.21.0', 'accelerate>=0.21.0'],
        # pip install "beyondfid[timm]"          → convnext, mae
        'timm': ['timm>=0.9.0'],
        # pip install "beyondfid[clip]"          → clip (git-only package)
        'clip': ['clip @ git+https://github.com/openai/CLIP.git'],
        # pip install "beyondfid[fld]"           → FLD metric (git-only package)
        'fld': ['fld @ git+https://github.com/marcojira/fld.git'],
        # pip install "beyondfid[video]"         → video preprocessing
        'video': ['av>=10.0.0'],
        # pip install "beyondfid[all]"           → everything
        'all': [
            'transformers>=4.30.0',
            'diffusers>=0.21.0',
            'accelerate>=0.21.0',
            'timm>=0.9.0',
            'clip @ git+https://github.com/openai/CLIP.git',
            'fld @ git+https://github.com/marcojira/fld.git',
            'av>=10.0.0',
        ],
    },
    author="Mischa Dombrowski",
    author_email="mischa.dombrowski@gmail.com",
    description="Python package to efficiently compute common metrics used for unconditional image generation",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    url="https://github.com/MischaD/BeyondFID",
    classifiers=[
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
    ],
    python_requires='>=3.8',
)