[Stable Diffusion]Segment Anything实现商业换装
发布时间:2024年06月06日
Segment Anything是一种基于深度学习算法的图像分割工具,它可以将图像当中的每一个像素分配到对应的区域当中,能够实现精准的边缘识别。与传统的分割工具相比,它的优势就在于自动检测,大大提高了分割效率。
官网:https://segment-anything.com/
1.Segment Anything安装
���安装方法:
1、在扩展面板中安装Segment Anything
地址:https://github.com/continue-revolution/sd-webui-segment-anything.git
2、guthub安装
guthub地址:https://github.com/continue-revolution/sd-webui-segment-anything
00001.
00002.
00003.
00004.
3、安装并重启之后,可以在面板中看到Segment Anything插件,此时还需要安装SAM模型。
在guthub上有三个SAM模型,分别是vit_h(2.56GB)、vit_l(1.25GB)、vit_b(375MB),越大的模型效果越好,但是越耗显存,我们下载vit_h就可以了。
4、segment-anything的SAM模型,我已经放在网盘了,下载后把SAM模型放在“……\sd-webui-aki-v4\extensions\sd-webui-segment-anything\models\sam”路径下。
00001.
链接:https://pan.quark.cn/s/e077a832ff39
.Segment Anything使用测试
左键点击图像添加一个黑色的正向标记点(想提取的部分)。右键点击图像添加一个红色的反向标记点(不想提取的部分)。左键再次点击可以删除标记点。
6、点击预览分离结果,可以看到SAM替我们分离出来的方案。
7、现在我想使用第一个蒙版来进行发型替换,所以我选择编号为0的蒙版。
8、点开“展开蒙版设置”,可以设定蒙版扩展量,就相当于是把蒙版扩展出一部分,让更换信息被充分包裹。默认是10,点击发送到“重绘蒙版”。
00001.
9.接下来,在图生图界面中,输入想替换的blue hair,在绘制区域选项中,选择仅蒙版。(刚才的图片已经进入“上传重绘蒙版”)
10、静静等待生成图片。
9、当然,我们也可以结合controlnet,如openpose来控制人物的身体,控制姿势等。
3.生成效果图
在生成的效果图中,可以看到人物发型已经替换过来了,我们完成了局部“换肤”。
出自:https://mp.weixin.qq.com/s/DMKXleMrEd1DROe6DvQmfw
Booth.ai,AI电商产品场景图生成器。