chenchaoyun commited on
Commit
4e14343
·
1 Parent(s): 832d159
Files changed (1) hide show
  1. config.py +1 -1
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", "yolov8n-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()
 
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()