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开启后,分割的速度会快些,但兼容性可能较差。