from setuptools import setup, find_packages

setup(
    name="uav_seg",
    version="0.1.0",
    packages=find_packages(),
    install_requires=[
        "torch>=1.7.0",
        "torchvision>=0.8.0",
        "numpy>=1.19.0",
        "pillow>=8.0.0",
        "scikit-learn>=0.24.0",
        "matplotlib>=3.3.0",
        "tqdm>=4.50.0",
        "rasterio>=1.2.0",  # 用于处理地理空间栅格数据
        "albumentations>=1.0.0",  # 用于数据增强
    ],
    author="UAV-Seg Team",
    author_email="example@example.com",
    description="用于大尺度遥感图像语义分割的Python库",
    long_description=open("README.md", encoding='utf-8').read(),
    long_description_content_type="text/markdown",
    url="https://github.com/example/uav_seg",
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    python_requires=">=3.7",
)