File size: 602 Bytes
e98bd8c 85ebba9 10e4799 85ebba9 e98bd8c |
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 |
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"
)
|