|
|
|
|
|
|
|
build:
|
|
gpu: true
|
|
system_packages:
|
|
- "libgl1-mesa-glx"
|
|
- "libglib2.0-0"
|
|
- "ffmpeg"
|
|
python_version: "3.11"
|
|
python_packages:
|
|
- "torch==2.0.1"
|
|
- "torchvision==0.15.2"
|
|
- "basicsr==1.4.2"
|
|
- "kornia==0.5.1"
|
|
- "face-alignment==1.3.4"
|
|
- "ninja==1.10.2.3"
|
|
- "einops==0.4.1"
|
|
- "facexlib==0.2.5"
|
|
- "librosa==0.9.2"
|
|
- "cmake==3.27.7"
|
|
- "numpy==1.23.4"
|
|
run:
|
|
- pip install dlib
|
|
- mkdir -p /root/.pyenv/versions/3.11.6/lib/python3.11/site-packages/facexlib/weights/ && wget --output-document "/root/.pyenv/versions/3.11.6/lib/python3.11/site-packages/facexlib/weights/detection_Resnet50_Final.pth" "https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth"
|
|
- mkdir -p /root/.pyenv/versions/3.11.6/lib/python3.11/site-packages/facexlib/weights/ && wget --output-document "/root/.pyenv/versions/3.11.6/lib/python3.11/site-packages/facexlib/weights/parsing_parsenet.pth" "https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth"
|
|
- mkdir -p /root/.cache/torch/hub/checkpoints/ && wget --output-document "/root/.cache/torch/hub/checkpoints/s3fd-619a316812.pth" "https://www.adrianbulat.com/downloads/python-fan/s3fd-619a316812.pth"
|
|
- mkdir -p /root/.cache/torch/hub/checkpoints/ && wget --output-document "/root/.cache/torch/hub/checkpoints/2DFAN4-cd938726ad.zip" "https://www.adrianbulat.com/downloads/python-fan/2DFAN4-cd938726ad.zip"
|
|
predict: "predict.py:Predictor"
|
|
|
