找回密码
 立即注册
查看: 695|回复: 0

AI抠图使用指南:Stable Diffusion WebUI Rembg实用技巧

[复制链接]

2

主题

3

回帖

33

积分

新手上路

积分
33
发表于 2023-8-7 16:17:13 | 显示全部楼层 |阅读模式
抠图是图像措置东西的一项必备能力,可以用在重绘、重组、改换布景等场景。比来我一直在探索 Stable Diffusion WebUI 的各项能力,那么 SD WebUI 的抠图能力表示如何呢?这篇文章就给大师分享一下。
安装插件

作为一个生成式AI,SD本身并没有精细的抠图控制能力,它需要借助插件来实现。
这里我们使用 stable-diffusion-webui-rembg 这个插件,插件安装成功后,会呈此刻“高清化”这个页面的底部。


安装方式一

适合网络访谒各种资源斗劲畅通的同学。
插件地址:https://github.com/AUTOMATIC1111/stable-diffusion-webui-rembg
直接在 Stable Diffusion WebUI 中通过网址安装,操作法式如下图所示:



第5点指示插件已经安装成功,只需要在“已安装”页面中重启SD WebUI就可以了。
实际抠图时,插件还会按照选择的抠图算法自动下载相关的模型,第一次使用某个算法抠图时等待的时间会长一点。
安装方式二

适合访谒外网不怎么便利的同学。
先通过此外东西,比如迅雷下载到当地,然后再部署到相应的目录。
插件地址:https://github.com/AUTOMATIC1111/stable-diffusion-webui-rembg
算法模型地址:https://github.com/danielgatis/rembg#models
或者下载我已经打包好的文件,关/注/公/众/号:萤火遛AI,动员静:抠图,即可获取相关文件。


stable-diffusion-webui-rembg.zip 是插件法式,解压后放到 stable-diffusion-webui/extensions 目录。
u2net.zip 是算法模型文件,解压后放到当前用户的临时目录:

  • Windows是:C:\Users\{你登录的Windows用户名}\.u2net
  • Linux是:/root/.u2net/
如果部署在云处事器,需要先上传zip压缩文件,再解压到指定的目录。参考解压指令:
unzip -o /root/autodl-tmp/stable-diffusion-webui-rembg.zip -d /root/stable-diffusion-webui/extensions/stable-diffusion-webui-rembg/
unzip -o /root/autodl-tmp/u2net.zip -d /root/.u2net/安装完毕之后,记得重启SD。
使用rembg

基本使用

点击“高清化”页签,上传要抠图的照片,“Remove background”这里选择抠图算法,最后点击“生成”。





看看默认的抠图效果:头发这里还有些底色没去掉,这部门和布景有些交错,算法不能很好的识别。
注意人像周围黑色的部门实际上都是透明的,我们把这张图片放到白色的布景中人像周围就是白色了。



我们把这个“Alpha matting”勾上,然后会出来几个选项,把“Erode size”的值改为“15”,它可以优化前景图边缘的分割。



从头生成,看看此次的成果,头发这块基本上看不出来之前的底色了,但是衣领这里有些模糊了。我测验考试了分歧的参数,也没能达到完美。或许把图片放到PS中再措置下是个正确的方式,以成果为导向,不能死抱着AI不放,有兴趣的可以尝尝。



参数介绍

上面做了一个基本的演示,但是这些参数我们如何配置才能阐扬最好的效果呢?这一小节就来看看这些参数的定义。
先看看这几个算法模型:

  • u2net:通用的的预训练模型,凡是用这个就行。
  • u2netp:u2net的轻量级版本。
  • u2net_human_seg:专门针对人像分割的预训练模型,只是分割人像时建议使用。
  • u2net_cloth_seg:专门从人像上抠衣服的预训练模型,它会把衣服分成三部门:上半身、下半身和全身。
  • silueta:和u2net不异,但是大小减少到43Mb,便利在小内存机器上使用。
  • isnet-general-use :一个新的通用的预训练模型。
  • isnet-anime:专门针对动画人物的高精度分割。
再看看它的两个选项:

  • Alpha matting:Alpha遮罩,这是一个斗劲专业的图像措置术语。在图像措置中,有一个东西用来暗示图像中每个像素点的透明度,这个东西称为Alpha通道;然后Alpha遮罩操作Alpha通道来控制图像的透明度,从而达到隐藏或显示某些部门的目的;在抠图这里就是努力让前景部门都显示出来,让布景部门都变透明。这个选项有三个参数,用来控制抠图的效果,我们看下:

    • Erode size:Alpha抠图腐蚀尺寸,通过在图像中构建一个长宽为这个值的矩形进行腐蚀。太小了前景和布景分手不彻底,边缘有交叉;太大了前景和布景会腐蚀的太多,边缘缺损明显。
    • Foreground threshold:前景图像的阈值,值过小布景可能被识别为前景,值过大前景可能被识别为布景。
    • Background threshold:布景图像的阈值,值小了前景可能被识别为布景,值大了布景可能识别为前景。

使用Alpha遮罩时可以初始用这几个经验值:(15, 220, 100),具体参数值再按照实际情况进行调整。

  • Return mask:返回抠图的蒙邦畿,下面顿时就会介绍它的用法和用途。
使用蒙版

这一小节以改换图片布景为例,演示蒙版的用法。具体想法是生成一张马斯克登陆火星遭遇外星人的照片。
生成蒙版

只要在生成时勾选上“Return mask”,最终输出图片就会变成蒙邦畿。



如下图所示,可以看到人物变成了白色的蒙版,我们把这个蒙邦畿片先保留到本机,后面顿时要用。



改换布景

在“图生图”中打开“局部绘制(上传蒙版)”。这里需要上传两张照片,一张原图,一张人物的蒙邦畿。



图生图的具体参数如下:
提示词:(the desert), ((night)), dim sun, (stargate), a man in a suit and white shirt smiling for a picture, a alien standing in the distance, digital painting, stargatejackal,surrealistic, hdri, smooth, sharp focus, illustration, fantasy, intricate, elegant, highly detailed, 8k <lora:sgasgard_v1:1>
反向提示词:EasyNegative, moon
缩放模式:填充,因为我想调整下图片的宽高比到 16:9,原图没这么宽,所以需要填充新扩展的空间。
蒙版模式:绘制非蒙版内容,在“局部绘制(上传蒙版)”中白色的区域是蒙版,因为我们要重绘布景,所以这里选择的是“绘制非蒙版内容”。
采样器:DPM++ 2M SDE Karras,这是比来新加的一个采样器,建议体验下。当然也可以用此外采样器。
采样步数:配合采样器设置。
宽度、高度:按照 16:9的比例设置。


看看出图的效果:


<hr/>以上就是本文的主要内容了,感兴趣的同学赶紧去尝尝吧。
如果你刚开始学习AI绘画,建议先看这两篇 Stable Diffusion WebUI 安装指南:
手把手教你在本机安装Stable Diffusion秋叶整合包
手把手教你在云环境炼丹(部署Stable Diffusion WebUI)

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

×
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Archiver|手机版|小黑屋|T9AI - 深度人工智能平台 ( 沪ICP备2023010006号 )

GMT+8, 2024-11-24 14:27 , Processed in 0.061159 second(s), 23 queries .

Powered by Discuz! X3.5

© 2001-2024 Discuz! Team.

快速回复 返回顶部 返回列表