ZhengPeng7
/

BiRefNet-DIS5K

Image Segmentation

background-removal

mask-generation

Dichotomous Image Segmentation

pytorch_model_hub_mixin

model_hub_mixin

Model card Files Files and versions Community

ZhengPeng7 commited on Sep 20

Commit

2ce7ba5

•

1 Parent(s): bcd88a0

Update birefnet.py

Files changed (1) hide show

birefnet.py +1 -1

birefnet.py CHANGED Viewed

@@ -7,7 +7,7 @@ import math
 class Config():
     def __init__(self) -> None:
         # PATH settings
-        self.sys_home_dir = os.environ['HOME']     # Make up your file system as: SYS_HOME_DIR/codes/dis/BiRefNet, SYS_HOME_DIR/datasets/dis/xx, SYS_HOME_DIR/weights/xx
         # TASK settings
         self.task = ['DIS5K', 'COD', 'HRSOD', 'DIS5K+HRSOD+HRS10K', 'P3M-10k'][0]

 class Config():
     def __init__(self) -> None:
         # PATH settings
+        self.sys_home_dir = os.path.expanduser('~')     # Make up your file system as: SYS_HOME_DIR/codes/dis/BiRefNet, SYS_HOME_DIR/datasets/dis/xx, SYS_HOME_DIR/weights/xx
         # TASK settings
         self.task = ['DIS5K', 'COD', 'HRSOD', 'DIS5K+HRSOD+HRS10K', 'P3M-10k'][0]