deepface/cog.yaml at main · sonigeez/deepface · GitHub

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
build:
  gpu: true
  python_version: "3.10"
  run:
    - pip uninstall onnxruntime onnxruntime-gpu -y
    - pip install onnxruntime-gpu==1.15.0
    - pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu118
  python_packages:
    - "opencv-contrib-python-headless==4.7.0.72"
    - "ftfy==6.1.1"
    - "scipy==1.9.3"
    - "Pillow==9.4.0"
    - "mediapipe"
    - "numpy==1.24.3"
    - "opencv-python-headless==4.7.0.72"
    - "onnx==1.14.0"
    - "insightface==0.7.3"
    - "psutil==5.9.5"
    - "tensorflow>=2.0.0"
    - "gfpgan"
  system_packages:
    - ffmpeg
    - libsm6
    - libxext6
predict: "predict.py:Predictor"