chenchaoyun
commited on
Commit
·
4e14343
1
Parent(s):
832d159
fix
Browse files
config.py
CHANGED
|
@@ -285,7 +285,7 @@ UPSCALE_SIZE = int(os.environ.get("UPSCALE_SIZE", 2))
|
|
| 285 |
SAVE_QUALITY = int(os.environ.get("SAVE_QUALITY", 90))
|
| 286 |
REALESRGAN_MODEL = os.environ.get("REALESRGAN_MODEL", "realesr-general-x4v3")
|
| 287 |
# yolov11n-face.pt / yolov8n-face.pt
|
| 288 |
-
YOLO_MODEL = os.environ.get("YOLO_MODEL", "
|
| 289 |
# mobilenetv3/resnet50
|
| 290 |
RVM_MODEL = os.environ.get("RVM_MODEL", "resnet50")
|
| 291 |
RVM_LOCAL_REPO = os.environ.get("RVM_LOCAL_REPO", "/opt/data/RobustVideoMatting").strip()
|
|
|
|
| 285 |
SAVE_QUALITY = int(os.environ.get("SAVE_QUALITY", 90))
|
| 286 |
REALESRGAN_MODEL = os.environ.get("REALESRGAN_MODEL", "realesr-general-x4v3")
|
| 287 |
# yolov11n-face.pt / yolov8n-face.pt
|
| 288 |
+
YOLO_MODEL = os.environ.get("YOLO_MODEL", "yolov11n-face.pt")
|
| 289 |
# mobilenetv3/resnet50
|
| 290 |
RVM_MODEL = os.environ.get("RVM_MODEL", "resnet50")
|
| 291 |
RVM_LOCAL_REPO = os.environ.get("RVM_LOCAL_REPO", "/opt/data/RobustVideoMatting").strip()
|