chenchaoyun commited on
Commit
0d0c76e
·
1 Parent(s): 3332c6d
Files changed (1) hide show
  1. 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.1 \
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=false \
34
- AUTO_INIT_DDCOLOR=false \
35
- AUTO_INIT_REALESRGAN=false \
36
- AUTO_INIT_ANIME_STYLE=false \
37
  AUTO_INIT_CLIP=false \
38
- AUTO_INIT_RVM=false \
39
- AUTO_INIT_REMBG=false \
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 \