chenchaoyun commited on
Commit ·
407b6c0
1
Parent(s): 229d96a
fix: web
Browse files- 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 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
FACE_CONFIDENCE=0.7 \
|
| 15 |
-
|
| 16 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
ENABLE_REALESRGAN=true \
|
| 18 |
-
ENABLE_ANIME_STYLE=true \
|
| 19 |
ENABLE_RVM=true \
|
| 20 |
ENABLE_REMBG=true \
|
|
|
|
|
|
|
| 21 |
ENABLE_CLIP=false \
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 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 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|