第二步SAM模型接入

README.md

@@ -8,7 +8,7 @@
 2. 使用 SAM 根据语义、粗框和程序候选生成 mask，经语义/几何检查后从原图提取可见像素。
 3. 使用图片编辑模型补全被遮挡的部件，再进行几何验收和 Spine 导出。
 
-目前第一阶段已经可以完整运行。SAM 3.1 checkpoint 路径和图片编辑适配器已经接入配置，但 SAM 推理、补全结果合并及新管线的 Spine 导出尚未完成。
+目前第一阶段和第二阶段均可运行。第二阶段在 `ai-spine` Conda 环境的临时子进程中加载本地 SAM 3.1 权重，使用 GPU 分割并生成半透明标注图、白底拆解图和透明部件。分割结果仍需语义复核；补全结果合并及新管线的 Spine 导出尚未完成。
 
 仓库仍保留旧版“AI 生成白底拆件图 → OpenCV 拆分 → Genielabs 定位 → Spine 导出”流程，用于对照和回归测试。
 
@@ -16,6 +16,7 @@
 
 - [项目修改约定](AGENTS.md)：每次修改都必须遵守，Python 固定使用 `ai-spine` Conda 环境。
 - [第一阶段：视觉 LLM 角色结构理解](docs/1-understanding-stage.md)：执行流程、代码对应关系、数据格式和输出文件。
+- [第二阶段：本地 SAM 分割](docs/2-segmentation-stage.md)：环境、运行命令、权重适配、输出和质量边界。
 - [模型阶段配置与实验说明](docs/model-stages.md)：模型切换、协议适配、图片编辑探针、缓存和错误处理。
 
 ## 环境准备
@@ -47,8 +48,9 @@ VECTOR_ENGINE_API_KEY=你的令牌
 SPINE_UNDERSTANDING_MODEL=deepseek-flash
 SPINE_COMPLETION_MODEL=qwen-image-3.0
 
-# 当前只记录 checkpoint；SAM 推理尚未接入
-SAM_MODEL_DIR=F:/Study/sam3.1
+# 本地 SAM 根目录和源码仓库；分割结束后子进程退出
+SAM_MODEL_DIR=F:/Study/SAM3
+SAM_CODE_DIR=F:/Study/SAM3/sam3
 ```
 
 模型 ID 会原样发送给中转站，不进行隐式别名转换。不要把 API Key 写进 JSON 配置或提交到版本库。
@@ -65,11 +67,13 @@ SAM_MODEL_DIR=F:/Study/sam3.1
 - `Pillow`、`numpy`、`opencv-python`；
 - 相邻目录中的 `spine-animation-ai` 项目。
 
-在 `AI_spine` 目录安装缺失的 Python 依赖（已满足版本要求的包会保留）：
+当前本机验证组合为 Python 3.11、PyTorch 2.7.1 + CUDA 12.6、torchvision 0.22.1。在 `AI_spine` 目录补全依赖；已有此 CUDA 版本时跳过第一条安装命令：
 
 ```powershell
 conda activate ai-spine
+python -m pip install torch==2.7.1 torchvision==0.22.1 --index-url https://download.pytorch.org/whl/cu126
 python -m pip install -r .\requirements.txt
+python -m pip install --no-deps --no-build-isolation -e F:\Study\SAM3\sam3
 ```
 
 不激活终端也可明确指定安装目标：
@@ -168,6 +172,22 @@ node .\pipeline.mjs --mode annotated-parts `
 
 每组模型实验应使用独立的输出目录，避免覆盖对照结果。只有输入、配置、代码和产物哈希全部一致时，才可以通过 `--resume` 复用结果。
 
+### 使用本地 SAM 生成标注和拆解图
+
+复用已有第一阶段结果，不重新请求 DeepSeek，也不需要中转站密钥：
+
+```powershell
+conda run --no-capture-output -n ai-spine node .\tools\model-check.mjs `
+  --stage segmentation `
+  --input-dir .\output\sombrero-grounding-v2 `
+  --out-dir .\output\sombrero-sam-local `
+  --sam-dir F:/Study/SAM3
+```
+
+打开输出目录的 `report.html`：`annotated.png` 是原图上的不同半透明颜色标注，`parts-sheet.png` 是白底拆解图，`parts/` 保存原像素透明部件。运行期间加载 GPU 权重，完成后子进程退出，不启动常驻服务。
+
+加入 `--dry-run` 可检查环境与本地路径，不加载权重推理。输入必须含 `source.png`、v2 `semantic-parts.json`、`segmentation-prompts.json`。完整参数与质量检查见[第二阶段文档](docs/2-segmentation-stage.md)。
+
 ### 测试图片补全接口
 
 此命令只验证图片编辑适配器并保存原始候选，不代表补全部件已经验收通过：