chenchaoyun commited on
Commit ·
0d0c76e
1
Parent(s): 3332c6d
fix
Browse files- Dockerfile +7 -7
Dockerfile
CHANGED
|
@@ -9,7 +9,7 @@ ENV TZ=Asia/Shanghai \
|
|
| 9 |
CELEBRITY_SOURCE_DIR=/opt/data/chinese_celeb_dataset \
|
| 10 |
GENDER_CONFIDENCE=1 \
|
| 11 |
UPSCALE_SIZE=2 \
|
| 12 |
-
AGE_CONFIDENCE=0.
|
| 13 |
DRAW_SCORE=true \
|
| 14 |
FACE_CONFIDENCE=0.7 \
|
| 15 |
ENABLE_DDCOLOR=true \
|
|
@@ -30,13 +30,13 @@ ENV TZ=Asia/Shanghai \
|
|
| 30 |
RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting \
|
| 31 |
RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth \
|
| 32 |
RVM_MODEL=resnet50 \
|
| 33 |
-
AUTO_INIT_GFPGAN=
|
| 34 |
-
AUTO_INIT_DDCOLOR=
|
| 35 |
-
AUTO_INIT_REALESRGAN=
|
| 36 |
-
AUTO_INIT_ANIME_STYLE=
|
| 37 |
AUTO_INIT_CLIP=false \
|
| 38 |
-
AUTO_INIT_RVM=
|
| 39 |
-
AUTO_INIT_REMBG=
|
| 40 |
ENABLE_WARMUP=true \
|
| 41 |
REALESRGAN_MODEL=realesr-general-x4v3 \
|
| 42 |
CELEBRITY_FIND_THRESHOLD=0.87 \
|
|
|
|
| 9 |
CELEBRITY_SOURCE_DIR=/opt/data/chinese_celeb_dataset \
|
| 10 |
GENDER_CONFIDENCE=1 \
|
| 11 |
UPSCALE_SIZE=2 \
|
| 12 |
+
AGE_CONFIDENCE=0.9999 \
|
| 13 |
DRAW_SCORE=true \
|
| 14 |
FACE_CONFIDENCE=0.7 \
|
| 15 |
ENABLE_DDCOLOR=true \
|
|
|
|
| 30 |
RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting \
|
| 31 |
RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth \
|
| 32 |
RVM_MODEL=resnet50 \
|
| 33 |
+
AUTO_INIT_GFPGAN=true \
|
| 34 |
+
AUTO_INIT_DDCOLOR=true \
|
| 35 |
+
AUTO_INIT_REALESRGAN=true \
|
| 36 |
+
AUTO_INIT_ANIME_STYLE=true \
|
| 37 |
AUTO_INIT_CLIP=false \
|
| 38 |
+
AUTO_INIT_RVM=true \
|
| 39 |
+
AUTO_INIT_REMBG=true \
|
| 40 |
ENABLE_WARMUP=true \
|
| 41 |
REALESRGAN_MODEL=realesr-general-x4v3 \
|
| 42 |
CELEBRITY_FIND_THRESHOLD=0.87 \
|