diff --git a/py/birefnet_ultra_v2.py b/py/birefnet_ultra_v2.py index 883b7e3..370ede1 100644 --- a/py/birefnet_ultra_v2.py +++ b/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 diff --git a/py/segformer_ultra.py b/py/segformer_ultra.py index 3d5b8e9..b8c97ae 100644 --- a/py/segformer_ultra.py +++ b/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':