|
|
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"] |
|
|
|
