IS-Net ONNX FP16 自动抠图模型
本仓库提供 isnet_fp16.onnx,用于图像前景分割与背景移除。模型根据输入图片预测前景概率遮罩,可作为人物、商品和其他主体抠图流程的基础,适合接入图片编辑器、素材处理工具或本地自动化工作流。
文件说明
| 文件 | 用途 | 大小 |
|---|---|---|
isnet_fp16.onnx |
IS-Net FP16 ONNX 模型 | 88,152,708 字节,约 88.15 MB |
FP16 表示此模型的半精度版本;应用接入文档使用的输入仍为 Float32,不能仅依据文件名将输入改成 Float16。
输入与输出
以下规范来自随模型提供的本地应用接入说明:
- 输入:RGB 图像,Float32,NCHW 排列,形状
[1, 3, 1024, 1024]。 - 预处理:将图片缩放至 1024 × 1024,按 RGB 通道排列,使用
(pixel - 128) / 256归一化;pixel 为 0–255 像素值。 - 输出:前景概率遮罩。应用需将遮罩缩放回原图尺寸,再将其用于透明度合成。
- 透明 PNG 的合成与导出由应用完成,模型本身不直接生成 PNG 文件。
运行与接入
可使用支持该 ONNX 计算图的运行时集成。原应用使用 ONNX Runtime Web,优先尝试具备 shader-f16 支持的 WebGPU,必要时回退到 WASM。实际运行兼容性取决于运行时版本、浏览器和设备,应在目标环境验证。
建议流程:读取图片 → RGB 缩放与归一化 → ONNX 推理 → 遮罩还原到原始尺寸 → Alpha 合成 → 导出结果。输入、输出节点名称请以实际模型元数据为准。
去色边、边缘收缩、透明度阈值和不同抠图预设属于应用层后处理,并非额外模型。
效果与限制
复杂背景、细发丝、半透明物体以及前景与背景颜色相近的图片可能出现漏分、残留背景或边缘瑕疵。必要时结合边缘后处理或人工修正。本仓库不承诺所有图片均可获得准确遮罩;上传过程未重新执行推理质量评测。
来源说明
本文件来自本地 Aarelier 参考项目所附模型,后被集成到 StarFlame 的抠图功能。此仓库用于保存和分发该 ONNX 文件,不代表 IS-Net 官方发布。
目前尚未核实该文件的原始训练权重版本、转换脚本、导出参数和具体许可证,因此未填写许可证标签,也不声明它与某个官方权重文件完全等价。使用、修改或再分发时应进一步核实原始授权条款。
文件校验
SHA256:
2eb4b5dda7ec41c617e59706e5aafa1f978c9a5f983d2518d9f0ae4d6eb04f20