微信小程序实现SD一键换装

前些天发了一条SD使用API换装的视频，有很多朋友关注，有的小伙伴表示使用API难度太大。我做了一个通过API实现SD换装的小程序，希望能够给大家带来启发。

首先简单分析一下实现思路。

1、  用API实现SD的基本功能

2、  用小程序将原始图片发送给服务器

3、  用分割模型自动分割出衣服蒙版

4、  将提示词输入给API，由服务器算出结果

5、  将换装后的图片返回给小程序并显示

下面我们来分步解释程序的实现

第一步，用小程序实现SD的基本功能，因为stable diffusion WEB UI 已经提供了API接口，所以小程序只要访问服务器对应地址即可实现SD的功能，这里我就不展开说明了，有兴趣的小伙伴可以查看我之前发的有关API的视频。

第二步，服务器接收来自小程序的原始图片，这里我使用阿里云的oss桶作为中转，可以很方便的将手机中的原始图片传递到服务器中。相关代码我会贴在文档后面。

第三步，使用分割模型对原始图片进行运算，得到衣服的蒙版。这部分比较复杂，需要使用专门的算法进行分割，还要合并类似的蒙版，这里我使用了一个判断语句，将蒙版类型进行筛选，然后将获得的蒙版合并，然后进行二值化，并提高蒙版图片的对比度，生成SD可用的蒙版图片。

第四步，将来自小程序输入端的提示词、原始图片和新生成的蒙版图片输入给SD的API，让服务器进行AI运算，获得换装后的图片。

最后一步，将服务器算好的图再通过阿里云的oss桶返回到小程序，小程序自动刷新就能看到换装后的图片了。

这种方式的优点是可以通过手机就能一键完成对图片的换装操作。缺点是：分割模型对原有服装分割的精度较高，换装范围仅限于原图衣服的范围，所以换装的拓展性有限制，不能超出原有衣服的范围。另外，目前衣服的lora太少了，希望有兴趣的小伙伴能提供更多衣服的lora。

由于每件衣服都可以生成，独一无二的，与衣服一致的lora，所以只要有lora就可以将模特的衣服换装成特定的样式，如果淘宝的服装店能提供其服装的lora，那么远程试衣将走进现实。