Commit
•
049b245
1
Parent(s):
d6b8eb4
Update birefnet.py (#10)
Browse files- Update birefnet.py (8d153a7f721e03f44b882fdedb3299069781c83e)
Co-authored-by: Lain <not-lain@users.noreply.huggingface.co>
- 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.
|
| 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]
|