jhj0517
commited on
Commit
•
2444752
1
Parent(s):
d1ad5e8
Add REAL ESRGAN model path
Browse files- modules/utils/paths.py +3 -0
modules/utils/paths.py
CHANGED
|
@@ -30,6 +30,9 @@ MODEL_ANIMAL_PATHS = {
|
|
| 30 |
# Just animal detection model not the face, needs better model
|
| 31 |
"yolo_v5s_animal_det": os.path.join(MODELS_ANIMAL_DIR, "yolo_v5s_animal_det.n2x")
|
| 32 |
}
|
|
|
|
|
|
|
|
|
|
| 33 |
MASK_TEMPLATES = os.path.join(PROJECT_ROOT_DIR, "modules", "utils", "resources", "mask_template.png")
|
| 34 |
I18N_YAML_PATH = os.path.join(PROJECT_ROOT_DIR, "i18n", "translation.yaml")
|
| 35 |
|
|
|
|
| 30 |
# Just animal detection model not the face, needs better model
|
| 31 |
"yolo_v5s_animal_det": os.path.join(MODELS_ANIMAL_DIR, "yolo_v5s_animal_det.n2x")
|
| 32 |
}
|
| 33 |
+
MODEL_REAL_ESRGAN_PATH = {
|
| 34 |
+
"realesr-general-x4v3": os.path.join(MODELS_REAL_ESRGAN_DIR, "realesr-general-x4v3.pth")
|
| 35 |
+
}
|
| 36 |
MASK_TEMPLATES = os.path.join(PROJECT_ROOT_DIR, "modules", "utils", "resources", "mask_template.png")
|
| 37 |
I18N_YAML_PATH = os.path.join(PROJECT_ROOT_DIR, "i18n", "translation.yaml")
|
| 38 |
|