第二步SAM模型接入

2026-09-18 20:29:29 +08:00
parent 1a7b2f14a9
commit f095c28858
16 changed files with 788 additions and 26 deletions
@@ -1,3 +1,22 @@
-Pillow>=10.0
-numpy>=1.26
-opencv-python>=4.9
+Pillow>=10.1
+numpy==1.26.4
+opencv-python==4.11.0.86
+# SAM runtime. Install the CUDA torch/torchvision wheels first as documented.
+torch==2.7.1
+torchvision==0.22.1
+setuptools==80.9.0
+wheel
+timm>=1.0.17
+tqdm
+ftfy==6.1.1
+regex
+iopath>=0.1.10
+typing_extensions
+huggingface_hub
+einops
+psutil
+scipy
+pycocotools
+triton-windows>=3.3,<3.4; platform_system == "Windows"
+triton>=3.3,<3.4; platform_system == "Linux"
+# SAM source is provided separately through --sam-code-dir / SAM_CODE_DIR.