[project]
name = "infrared-data"
version = "0.2.0"
description = "Infrared Equipment Detection based on YOLO - New Dataset"
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
    "matplotlib>=3.10.8",
    "numpy>=2.2.6",
    "opencv-python>=4.13.0.92",
    "pandas>=2.3.3",
    "pillow>=12.1.1",
    "scikit-learn>=1.7.2",
    "seaborn>=0.13.2",
    "torch>=2.5.1",
    "torchvision>=0.20.1",
    "tqdm>=4.67.3",
    "ultralytics>=8.4.27",
]


[tool.setuptools.packages]
find = {where = ["src"]}

# PyTorch CUDA版本配置
[tool.uv]
# 指定PyTorch使用CUDA索引
[[tool.uv.index]]
name = "pytorch-cu130"
url = "https://download.pytorch.org/whl/cu130"
explicit = true

# torch和torchvision从PyTorch索引安装
[tool.uv.sources]
torch = [{ index = "pytorch-cu130" }]
torchvision = [{ index = "pytorch-cu130" }]