AI完美去水印工作流

财富   2024-12-22 09:38   山东  

今天推荐一个工作流,使用AI大模型去水印,效果更完美。

基本原理是先使用大模型识别到水印位置,然后利用AI绘画的局部重绘功能进行重绘,效果要比普通的PS更精致。

一、模型安装(下载见文末)

这个工作流要用到四组模型:

1、SAM模型,共有7个,模型越大识别和处理效果就越好,但对显存要求也比较高。我是12G的3060,用1.25G那个模型勉强可以(再大就容易爆显存)。

如果大模型对水印的识别不够精准,可以再此调整阈值。默认0.3,可以在0.21-0.29之间尝试。

安装路径:

ComfyUI\models\sams


2、GroundingDINO模型

作为当前性能最好的开集检测模型,Grounding DINO可以帮助构建海量的具有物体级别语义信息的多模态数据,从而有效地助力多模态大模型的训练。它可以将长文本描述中的短语与图像中的具体对象或场景精确匹配,以增强AI对视觉内容和文本之间关系的理解。

当然了,显存拮据的用小一点的。

安装路径:

ComfyUI\models\grounding-dino

3、局部重绘的head和patch两个模型

安装路径:

ComfyUI\models\inpaint

二、主要节点介绍:

其实这个工作流主要就两个节点,外加一个SDXL的大模型(随便一个常用的即可)。

一个负责识别和遮罩,一个负责对遮罩进行重绘。

三、工作流和模型下载

https://pan.quark.cn/s/d8aecff3f5cb




路过银河AI
凝视深渊过久,深渊将回以凝视。天空的深渊,便是天渊。
 最新文章