添加关闭遮罩功能,更灵活的处理透明和非透明图片

This commit is contained in:
Dontdrunk
2025-07-27 10:55:50 +08:00
parent baca988c6f
commit 2a09d97633
3 changed files with 11 additions and 9 deletions
+3 -3
View File
@@ -315,9 +315,9 @@
"name": "Image",
"tooltip": "Input image to add stroke effect"
},
"反转遮罩": {
"name": "Invert Mask",
"tooltip": "Whether to invert mask effect, enabled by default"
"关闭遮罩": {
"name": "Disable Mask",
"tooltip": "Disable mask input when enabled"
},
"位置": {
"name": "Position",
+3 -3
View File
@@ -315,9 +315,9 @@
"name": "图片",
"tooltip": "需要添加描边的输入图片"
},
"反转遮罩": {
"name": "反转遮罩",
"tooltip": "是否反转遮罩效果,默认开启"
"关闭遮罩": {
"name": "关闭遮罩",
"tooltip": "开启后将关闭遮罩输入"
},
"位置": {
"name": "位置",
+5 -3
View File
@@ -14,6 +14,7 @@ class DDImageStroke:
return {
"required": {
"图片": ("IMAGE",),
"关闭遮罩": ("BOOLEAN", {"default": False}),
"位置": (["外描边", "内描边", "居中描边"], {"default": "外描边"}),
"大小": ("INT", {"default": 5, "min": 1, "max": 200, "step": 1}),
"不透明度": ("INT", {"default": 100, "min": 1, "max": 100, "step": 1}),
@@ -249,12 +250,13 @@ class DDImageStroke:
return result
def add_stroke(self, 图片, 位置, 大小, 不透明度, 描边颜色, 遮罩=None):
def add_stroke(self, 图片, 关闭遮罩, 位置, 大小, 不透明度, 描边颜色, 遮罩=None):
"""
为图片添加描边效果
Args:
图片: 输入图片tensor
关闭遮罩: 是否关闭遮罩功能(True时忽略遮罩输入)
位置: 描边位置("外描边", "内描边", "居中描边")
大小: 描边大小(像素)
不透明度: 描边不透明度(1-100)
@@ -270,9 +272,9 @@ class DDImageStroke:
results = []
for i in range(batch_size):
# 获取对应的遮罩(如果有)
# 获取对应的遮罩(如果有且未关闭遮罩功能)
current_mask = None
if 遮罩 is not None:
if not 关闭遮罩 and 遮罩 is not None:
if len(遮罩.shape) == 3 and 遮罩.shape[0] > i:
current_mask = 遮罩[i:i+1]
elif len(遮罩.shape) == 2: