# setup.py
from setuptools import setup, find_packages

setup(
    name="SemanticModel",
    version="0.1.0", 
    description="Deep learning framework for semantic segmentation",
    author="Your Name",
    packages=find_packages(),
    python_requires=">=3.8",
    install_requires=[
        'torch',
        'torchvision',
        'tensorboard',
        'pyproj',
        'fiona==1.8.20',
        'rtree',
        'geopandas',
        'rasterio',
        'slidingwindow',
        'opencv-python',
        'wandb',
        'tifffile',
        'imagecodecs', 
        'albumentations',
        'segmentation-models-pytorch>=0.3.3'
    ],
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3.8",
    ],
)