[project]
name = "code"
version = "0.1.0"
description = "LVA needle tracking using deep learning"
readme = "README.md"
requires-python = ">=3.14"
dependencies = [
    "torch==2.10.0+cu130",
    "torchvision==0.25.0+cu130",
    "opencv-python",
    "matplotlib",
]

[tool.uv.sources]
torch = { index = "torch-cuda" }
torchvision = { index = "torch-cuda" }

[[tool.uv.index]]
name = "torch-cuda"
url = "https://download.pytorch.org/whl/cu130"
explicit = true