| from setuptools import find_packages, setup | |
| setup( | |
| name="segformer_plusplus", | |
| version="0.2", | |
| author="Marco Kantonis", | |
| description="Segformer++: Efficient Token-Merging Strategies for High-Resolution Semantic Segmentation", | |
| install_requires=[ | |
| 'torch>=2.0.1', | |
| 'tomesd', | |
| 'omegaconf', | |
| 'pyyaml', | |
| 'numpy', | |
| 'rich', | |
| 'yapf', | |
| 'addict', | |
| 'tqdm', | |
| 'packaging', | |
| 'Pillow', | |
| 'torchvision' | |
| ], | |
| packages=find_packages(), | |
| license='MIT', | |
| long_description="https://arxiv.org/abs/2405.14467" | |
| ) | |