Image Remove Bg

节点功能:用于图像背景移除。

输入参数

参数名称说明
images输入图像,支持多张。

输出参数

参数名称说明
image移除背景后的图像(支持透明通道)。
mask背景遮罩图(白色为前景)。

控件参数

参数名称说明
rem_mode背景移除模型,支持 RMBG-2.0、RMBG-1.4、Inspyrenet、BEN2。
image_output输出方式,控制是否预览、保存或隐藏图像。
save_prefix保存图像的文件名前缀。
torchscript_jit(Inspyrenet 专用)是否使用 JIT 加速模型。
add_background为结果添加背景(none/white/black)。
refine_foreground(BEN2 专用)是否启用前景精细化处理可在边缘处理更干净时启用,建议开启以提高质量。

这里提供了4个可选的分割模型,对于一些场景不复杂的图片,这四个模型分割效果差不多。

这里没有使用RMBG-1.4进行对比,是因为RMBG-2.0是它的进阶版,效果肯定更好一些。从这几个分割效果来看,对于复杂背景的分割,RMBG模型表现不好,对于半透明场景,BEN2表现不好。

可以对分割背景添加颜色,不过只有黑白两种颜色

torchscript_jit开启后,分割的速度会快些,但兼容性可能较差。