fix bug of rmbg node

This commit is contained in:
macro chen
2024-12-05 17:53:25 +08:00
parent 13acd2532a
commit 4dd89369b5
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
import torch
from PIL import Image
from .imagefunc import log, tensor2pil, pil2tensor, image2mask, RMBG, RGB2RGBA, mask_edge_detail
from .imagefunc import log, tensor2pil, pil2tensor, image2mask, mask2image, RMBG, RGB2RGBA, mask_edge_detail
class RemBgUltra:
+1 -1
View File
@@ -1,6 +1,6 @@
import torch
from PIL import Image
from .imagefunc import log, tensor2pil, pil2tensor, image2mask, RMBG, RGB2RGBA, mask_edge_detail
from .imagefunc import log, tensor2pil, pil2tensor, image2mask, mask2image, RMBG, RGB2RGBA, mask_edge_detail
from .imagefunc import guided_filter_alpha, histogram_remap, generate_VITMatte, generate_VITMatte_trimap