ZhengPeng7 commited on
Commit
1324e41
1 Parent(s): 29d3b6e

Update birefnet.py

Browse files
Files changed (1) hide show
  1. birefnet.py +1 -1
birefnet.py CHANGED
@@ -7,7 +7,7 @@ import math
7
  class Config():
8
  def __init__(self) -> None:
9
  # PATH settings
10
- 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
11
 
12
  # TASK settings
13
  self.task = ['DIS5K', 'COD', 'HRSOD', 'DIS5K+HRSOD+HRS10K', 'P3M-10k'][0]
 
7
  class Config():
8
  def __init__(self) -> None:
9
  # PATH settings
10
+ 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
11
 
12
  # TASK settings
13
  self.task = ['DIS5K', 'COD', 'HRSOD', 'DIS5K+HRSOD+HRS10K', 'P3M-10k'][0]