init

Dockerfile

@@ -27,7 +27,7 @@ ENV TZ=Asia/Shanghai \
     ENABLE_ANIME_PRELOAD=true \
     ENABLE_LOGGING=true \
     BEAUTY_ADJUST_ENABLED=true \
-    RVM_LOCAL_REPO=/opt/data/RobustVideoMatting \
+    RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting \
     RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth \
     RVM_MODEL=resnet50 \
     AUTO_INIT_GFPGAN=true \

config.py

@@ -296,8 +296,8 @@ REALESRGAN_MODEL = os.environ.get("REALESRGAN_MODEL", "realesr-general-x4v3")
 YOLO_MODEL = os.environ.get("YOLO_MODEL", "yolov8n-face.pt")
 # mobilenetv3/resnet50
 RVM_MODEL = os.environ.get("RVM_MODEL", "resnet50")
-RVM_LOCAL_REPO = os.environ.get("RVM_LOCAL_REPO", "").strip()
-RVM_WEIGHTS_PATH = os.environ.get("RVM_WEIGHTS_PATH", "").strip()
+RVM_LOCAL_REPO = os.environ.get("RVM_LOCAL_REPO", "/opt/data/RobustVideoMatting").strip()
+RVM_WEIGHTS_PATH = os.environ.get("RVM_WEIGHTS_PATH", "/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth").strip()
 DRAW_SCORE = os.environ.get("DRAW_SCORE", "true").lower() in ("1", "true", "on")
 
 # 颜值评分温和提升配置（默认开启；默认区间与力度：区间=[6.0, 8.0]，gamma=0.3）

start_local.sh

@@ -7,6 +7,9 @@ export AUTO_INIT_REALESRGAN=false
 export AUTO_INIT_ANIME_STYLE=false
 export AUTO_INIT_CLIP=false
 export AUTO_INIT_RVM=true
+export RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting
+export RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth
+export RVM_MODEL=resnet50
 export AUTO_INIT_REMBG=false
 uvicorn app:app --workers 1 --loop asyncio --http httptools --host 0.0.0.0 --port 7860 --timeout-keep-alive 600