nodes/kwaroran-abg-comfyui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[fix] misspelling

Browse Source
This commit is contained in:
kwaroran
2023-07-31 11:26:43 +09:00
parent 5ea07f07e6
commit 29d986f221
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+4 -4
View File
@@ -35,7 +35,7 @@ def rmbg_fn(img):
mask = mask.repeat(3, axis=2) mask = mask.repeat(3, axis=2)
return img return img
class RemoveImageBackgroundARB: class RemoveImageBackgroundabg:
def __init__(self): def __init__(self):
pass pass
@@ -48,10 +48,10 @@ class RemoveImageBackgroundARB:
} }
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE",)
FUNCTION = "agb_remover" FUNCTION = "abg_remover"
CATEGORY = "image" CATEGORY = "image"
def agb_remover(self, image:torch.Tensor): def abg_remover(self, image:torch.Tensor):
npa = image2nparray(image) npa = image2nparray(image)
print(npa.ndim) print(npa.ndim)
rmb = rmbg_fn(npa) rmb = rmbg_fn(npa)
@@ -76,5 +76,5 @@ def nparray2image(narray:np.array):
return tensor return tensor
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"Remove Image Background (AGB)": RemoveImageBackgroundARB "Remove Image Background (abg)": RemoveImageBackgroundabg
} }