File size: 2,261 Bytes
cd5aabe fae1594 cd5aabe 7af1f43 cd5aabe 12a0952 cd5aabe 613a82b cd5aabe c6e4a1e fae1594 cd5aabe a1be5bb cd5aabe a1be5bb 12a0952 cd5aabe fae1594 cd5aabe 7396c20 cd5aabe b47452e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
FROM python:3.10-slim
ENV TZ=Asia/Shanghai \
OUTPUT_DIR=/opt/data/output \
IMAGES_DIR=/opt/data/images \
MODELS_PATH=/opt/data/models \
DEEPFACE_HOME=/opt/data/models \
FAISS_INDEX_DIR=/opt/data/faiss \
CELEBRITY_SOURCE_DIR=/opt/data/chinese_celeb_dataset \
GENDER_CONFIDENCE=1 \
UPSCALE_SIZE=2 \
AGE_CONFIDENCE=0.1 \
DRAW_SCORE=true \
FACE_CONFIDENCE=0.7 \
ENABLE_DDCOLOR=true \
ENABLE_GFPGAN=true \
ENABLE_REALESRGAN=true \
ENABLE_ANIME_STYLE=true \
ENABLE_RVM=true \
ENABLE_REMBG=true \
ENABLE_CLIP=false \
CLEANUP_INTERVAL_HOURS=1 \
CLEANUP_AGE_HOURS=1 \
BEAUTY_ADJUST_GAMMA=0.8 \
BEAUTY_ADJUST_MIN=1.0 \
BEAUTY_ADJUST_MAX=9.0 \
ENABLE_ANIME_PRELOAD=false \
ENABLE_LOGGING=true \
BEAUTY_ADJUST_ENABLED=true \
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=false \
AUTO_INIT_DDCOLOR=false \
AUTO_INIT_REALESRGAN=false \
AUTO_INIT_ANIME_STYLE=false \
AUTO_INIT_CLIP=false \
AUTO_INIT_RVM=false \
AUTO_INIT_REMBG=false \
ENABLE_WARMUP=true \
REALESRGAN_MODEL=realesr-general-x4v3 \
CELEBRITY_FIND_THRESHOLD=0.87 \
FEMALE_AGE_ADJUSTMENT=4 \
HOSTNAME=HG
RUN mkdir -p /opt/data/chinese_celeb_dataset /opt/data/faiss /opt/data/models /opt/data/images /opt/data/output
WORKDIR /app
COPY requirements.txt .
COPY *.py /app/
COPY facelist-web /app/facelist-web
# 安装必要的系统工具和依赖
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
cmake \
git \
wget \
curl \
ca-certificates \
libopenblas-dev \
liblapack-dev \
libx11-dev \
libgtk-3-dev \
libboost-python-dev \
libglib2.0-0 \
libsm6 \
libxext6 \
libxrender-dev \
libgomp1 \
&& rm -rf /var/lib/apt/lists/*
RUN pip install --upgrade pip
# 安装所有依赖 - 现在可以一次性完成
RUN pip install --no-cache-dir -r requirements.txt
EXPOSE 7860
CMD ["uvicorn", "app:app", "--workers", "1", "--loop", "asyncio", "--http", "httptools", "--host", "0.0.0.0", "--port", "7860", "--timeout-keep-alive", "600"]
|