From 29d986f22179ef6fbccf07a3b536f187dbcd7045 Mon Sep 17 00:00:00 2001 From: kwaroran Date: Mon, 31 Jul 2023 11:26:43 +0900 Subject: [PATCH] [fix] misspelling --- __init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/__init__.py b/__init__.py index 4947104..65d41e0 100644 --- a/__init__.py +++ b/__init__.py @@ -35,7 +35,7 @@ def rmbg_fn(img): mask = mask.repeat(3, axis=2) return img -class RemoveImageBackgroundARB: +class RemoveImageBackgroundabg: def __init__(self): pass @@ -48,10 +48,10 @@ class RemoveImageBackgroundARB: } RETURN_TYPES = ("IMAGE",) - FUNCTION = "agb_remover" + FUNCTION = "abg_remover" CATEGORY = "image" - def agb_remover(self, image:torch.Tensor): + def abg_remover(self, image:torch.Tensor): npa = image2nparray(image) print(npa.ndim) rmb = rmbg_fn(npa) @@ -76,5 +76,5 @@ def nparray2image(narray:np.array): return tensor NODE_CLASS_MAPPINGS = { - "Remove Image Background (AGB)": RemoveImageBackgroundARB + "Remove Image Background (abg)": RemoveImageBackgroundabg } \ No newline at end of file