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
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support