SlotDiffusion/setup.py at main · Wuziyi616/SlotDiffusion · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from setuptools import setup, find_packages

requirements = [
    'numpy', 'pyyaml', 'six', 'tqdm', 'opencv-python', 'matplotlib', 'open3d',
    'pycocotools', 'pandas', 'moviepy', 'imageio', 'scikit-image', 'wandb',
    'torch', 'torchvision', 'torchmetrics', 'lpips', 'einops', 'transformers',
    'pytorch-lightning', 'nerv'
]

title = "SlotDiffusion: Object-Centric Learning with Diffusion Models"
setup(
    name="slotdiffusion",
    version='0.1.0',
    description=title,
    long_description=title,
    author="Ziyi Wu",
    author_email="ziyiwu@cs.toronto.edu",
    license="MIT",
    url="https://slotdiffusion.github.io/",
    keywords="object-centric diffusion model",
    packages=find_packages(),
    install_requires=requirements,
)