chenchaoyun commited on
Commit
407b6c0
·
1 Parent(s): 229d96a
Files changed (1) hide show
  1. Dockerfile +49 -26
Dockerfile CHANGED
@@ -1,47 +1,70 @@
1
  FROM python:3.10-slim
2
 
 
3
  ENV TZ=Asia/Shanghai \
 
4
  OUTPUT_DIR=/opt/data/output \
5
  IMAGES_DIR=/opt/data/images \
6
  MODELS_PATH=/opt/data/models \
7
  DEEPFACE_HOME=/opt/data/models \
8
  FAISS_INDEX_DIR=/opt/data/faiss \
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 \
16
- ENABLE_GFPGAN=true \
 
 
 
 
 
17
  ENABLE_REALESRGAN=true \
18
- ENABLE_ANIME_STYLE=true \
19
  ENABLE_RVM=true \
20
  ENABLE_REMBG=true \
 
 
21
  ENABLE_CLIP=false \
22
- CLEANUP_INTERVAL_HOURS=1 \
23
- CLEANUP_AGE_HOURS=1 \
24
- BEAUTY_ADJUST_GAMMA=0.8 \
25
- BEAUTY_ADJUST_MIN=1.0 \
26
- BEAUTY_ADJUST_MAX=9.0 \
27
- ENABLE_ANIME_PRELOAD=false \
28
- ENABLE_LOGGING=true \
29
- BEAUTY_ADJUST_ENABLED=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=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 \
43
- FEMALE_AGE_ADJUSTMENT=4 \
44
- HOSTNAME=HG
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
 
46
  RUN mkdir -p /opt/data/chinese_celeb_dataset /opt/data/faiss /opt/data/models /opt/data/images /opt/data/output
47
  WORKDIR /app
 
1
  FROM python:3.10-slim
2
 
3
+ # --- 系统与基础路径配置 ---
4
  ENV TZ=Asia/Shanghai \
5
+ HOSTNAME=claw \
6
  OUTPUT_DIR=/opt/data/output \
7
  IMAGES_DIR=/opt/data/images \
8
  MODELS_PATH=/opt/data/models \
9
  DEEPFACE_HOME=/opt/data/models \
10
  FAISS_INDEX_DIR=/opt/data/faiss \
11
+ CELEBRITY_SOURCE_DIR=/opt/data/chinese_celeb_dataset
12
+
13
+ # --- 算法逻辑与置信度阈值 ---
14
+ ENV GENDER_CONFIDENCE=0.9999 \
15
+ AGE_CONFIDENCE=0.9999 \
16
  FACE_CONFIDENCE=0.7 \
17
+ CELEBRITY_FIND_THRESHOLD=0.87 \
18
+ FEMALE_AGE_ADJUSTMENT=4 \
19
+ UPSCALE_SIZE=2 \
20
+ DRAW_SCORE=true
21
+
22
+ # --- AI 模型开关 (已开启 DDColor 与 Anime) ---
23
+ ENV ENABLE_GFPGAN=true \
24
  ENABLE_REALESRGAN=true \
 
25
  ENABLE_RVM=true \
26
  ENABLE_REMBG=true \
27
+ ENABLE_DDCOLOR=true \
28
+ ENABLE_ANIME_STYLE=true \
29
  ENABLE_CLIP=false \
30
+ ENABLE_WARMUP=true \
31
+ ENABLE_LOGGING=true
32
+
33
+ # --- 自动初始化设置 (全手动受控加载) ---
34
+ ENV AUTO_INIT_GFPGAN=false \
 
 
 
 
 
 
 
35
  AUTO_INIT_DDCOLOR=false \
36
  AUTO_INIT_REALESRGAN=false \
37
  AUTO_INIT_ANIME_STYLE=false \
38
  AUTO_INIT_CLIP=false \
39
  AUTO_INIT_RVM=false \
40
+ AUTO_INIT_REMBG=false
41
+
42
+ # --- 特定模型配置 ---
43
+ ENV REALESRGAN_MODEL=realesr-general-x4v3 \
44
+ RVM_MODEL=resnet50 \
45
+ RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting \
46
+ RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth \
47
+ HUGGINGFACE_SYNC_ENABLED=true
48
+
49
+ # --- 颜值评分与美化参数 ---
50
+ ENV BEAUTY_ADJUST_ENABLED=true \
51
+ BEAUTY_ADJUST_GAMMA=0.8 \
52
+ BEAUTY_ADJUST_MIN=1.0 \
53
+ BEAUTY_ADJUST_MAX=9.0
54
+
55
+ # --- 数据库配置 (MySQL) ---
56
+ ENV MYSQL_PORT=3306 \
57
+
58
+ # --- 对象存储配置 (Cloudflare R2 / BOS) ---
59
+ ENV BOS_IMAGE_DIR=images \
60
+ BOS_MODELS_PREFIX=models/s3 \
61
+ BOS_CELEBRITY_PREFIX=chinese_celeb_dataset \
62
+ BOS_UPLOAD_ENABLED=true
63
+
64
+ # --- 维护与清理 ---
65
+ ENV CLEANUP_INTERVAL_HOURS=24 \
66
+ CLEANUP_AGE_HOURS=24 \
67
+ BOS_CLEANUP_ENABLED=false
68
 
69
  RUN mkdir -p /opt/data/chinese_celeb_dataset /opt/data/faiss /opt/data/models /opt/data/images /opt/data/output
70
  WORKDIR /app