add packages

2026-08-07 16:52:10 +08:00
parent 2b80e3d380
commit c71e0be5cc
5 changed files with 761 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,7 +7,23 @@ authors = [
    { name = "Kunpeng", email = "mx18782708114@163.com" }
 ]
 requires-python = ">=3.14"
-dependencies = []
+dependencies = [
+    "matplotlib>=3.11.1",
+    "numpy>=2.5.1",
+    "opencv-python>=5.0.0.93",
+    "pandas>=3.0.5",
+    "torch>=2.13.0",
+    "torchvision>=0.28.0",
+]
+
+[tool.uv.sources]
+torch = { index = "pytorch-cu132" }
+torchvision = { index = "pytorch-cu132" }
+
+[[tool.uv.index]]
+name = "pytorch-cu132"
+url = "https://download.pytorch.org/whl/cu132"
+explicit = true

 [project.scripts]
 yolo = "yolo:main"