使用SAM(Segment Anything Model)去除图像背景,提取前景主体为透明PNG。适用于需要抠图、去除背景的场景。
SAM 背景清除是一项面向实际任务的技能,主要用于使用 Meta 的分块任意模式从图像中提取前景主题, 输出透明的 PNG;@% 快速启动;
该技能适合需要稳定复用相关能力的场景,可作为自动化工作流的一部分,也便于后续检查、调整和扩展。从功能定位来看,该技能强调把分散的操作要求整理成清晰、可复用的处理流程,使用户能够围绕既定目标快速准备输入、选择执行方式并获得结构化结果。实际使用前应先确认任务范围、数据来源、运行环境、必要权限和关键参数,再依据技能说明逐步执行;
若输入条件不完整,应先补齐信息或采用保守配置,避免因错误假设导致结果偏离需求。执行过程中需要关注工具调用是否成功、接口或依赖是否可用、输出格式是否符合预期,并对异常提示、缺失字段和边界情况进行处理;涉及批量任务时,还应保存进度,避免中断后重复操作。
使用 Meta 的 Segment Anything Model(SAM)从图像中提取前景主体,并输出带透明背景的 PNG 图像。
python3 scripts/segment.py
默认以图像中心作为前景提示点——适用于主体居中的肖像照和产品图。
| 参数 | 说明 | 默认值 |
|---|---|---|
input |
输入图像路径 | 必填 |
output |
输出 PNG 路径(单图模式)或输出目录(--all 模式) |
必填 |
--model |
模型尺寸:vit_b(速度快) · vit_l(中等) · vit_h(质量最佳) |
vit_h |
--checkpoint |
本地检查点路径;若未指定则自动下载 | auto |
--points |
前景提示点坐标,格式为 x,y,支持多个点 |
center |
--all |
网格扫描模式:提取图像中所有独立前景元素 | 关闭 |
--grid |
--all 模式下的网格密度;例如 16 表示 16×16 = 256 个探测点 |
16 |
--iou-thresh |
接受掩码所需的最小预测 IoU 值(仅 --all 模式生效) |
0.88 |
--min-area |
掩码最小面积占整图比例(仅 --all 模式生效) |
0.001 |
# 基础背景去除(自动下载 vit_h 模型,约 2.5GB) python3 scripts/segment.py photo.jpg output.png # 当主体偏离中心时,手动指定提示点 python3 scripts/segment.py photo.jpg output.png --points 320,240 # 使用轻量级模型并提供多个提示点 python3 scripts/segment.py photo.jpg output.png --model vit_b --points 320,240 400,300 # 提取全部前景元素(每个元素单独保存为一张 PNG) python3 scripts/segment.py photo.jpg ./elements/ --all # 使用更密集的网格以捕获小尺寸物体 python3 scripts/segment.py photo.jpg ./elements/ --all --grid 32 # 使用本地检查点文件 python3 scripts/segment.py photo.jpg output.png --checkpoint /path/to/sam_vit_h_4b8939.pth
segment_anything 库将在首次运行时自动安装;也可手动安装:
pip install git+https://github.com/facebookresearch/segment-anything.git pip install pillow numpy torch torchvision
~/.cache/sam/| 模型 | 大小 | 速度 | 质量 |
|---|---|---|---|
vit_b |
约 375 MB | 最快 | 良好 |
vit_l |
约 1.25 GB | 中等 | 更优 |
vit_h |
约 2.5 GB | 较慢 | 最佳 |
当系统中存在 GPU 时,CUDA 将自动启用。
相关专题
热门下载
相关下载
精品课程
共6课时 | 54.7万人学习
共89课时 | 134.1万人学习
共49课时 | 82.6万人学习