fix bug of BiRefNetUltraV2 and SegformerUltraV2

py/birefnet_ultra_v2.py

@@ -5,7 +5,7 @@ from torchvision import transforms
 import tqdm
 from .imagefunc import *
 from comfy.utils import ProgressBar
-sys.path.append(os.path.join(os.path.dirname(__file__), 'BiRefNet'))
+# sys.path.append(os.path.join(os.path.dirname(__file__), 'BiRefNet'))
 from .BiRefNet.models.birefnet import BiRefNet
 
 def get_models():
@@ -121,7 +121,7 @@ class LS_BiRefNetUltraV2:
             resize_sampler = Image.BILINEAR
             _mask = _mask.resize(orig_image.size, resize_sampler)
             brightness_image = ImageEnhance.Brightness(_mask)
-            _mask = brightness_image.enhance(factor=1.01)
+            _mask = brightness_image.enhance(factor=1.08)
             _mask = image2mask(_mask)
 
             detail_range = detail_erode + detail_dilate

py/segformer_ultra.py

@@ -486,9 +486,9 @@ class SegformerUltraV2:
             mask_image = Image.fromarray((1 - mask) * 255)
             mask_image = mask_image.convert("L")
             brightness_image = ImageEnhance.Brightness(mask_image)
-            mask_image = brightness_image.enhance(factor=1.01)
+            mask_image = brightness_image.enhance(factor=1.08)
             _mask = pil2tensor(mask_image)
-            
+
             detail_range = detail_erode + detail_dilate
             if process_detail:
                 if detail_method == 'GuidedFilter':