• Face X
• • Introduction
  • Demo
  • Usage
  • • Install
    • Usage
  • Models
  • • List
    • Download
    • Upload
  • Reference

Face X

Introduction

Face toolkits for detection, recognition based on onnxruntime

• face detection (bbox + kps5)
• face alignment (landmark68 + pitch, yaw, roll pose)
• face attribute (gender + age)
• face recognition (embedding)
• face parseing (19 labels)
• face restoration (codeformer 512)
• face makeup transfer (beautygan)
• face swap (inswapper 128)

Demo

face detection bbox kps5 landmark68 pose

face align parsing restoration

face makeup transfer & restoration

face swap

Usage

Install

# 
pip install git+https://openi.pcl.ac.cn/crj1998/face_toolkit.git

git clone https://openi.pcl.ac.cn/crj1998/face_toolkit.git && cd face_toolkit && pip install -e .

Usage

import cv2
import numpy as np
from facex import FaceX
from facex.model_zoo import get_model
from facex.face_align import norm_crop

app = FaceX(
    providers=[
        'CUDAExecutionProvider',
        'CPUExecutionProvider',
    ], 
    tasks=[
        "detection", 
        "landmark_3d_68", 
        "genderage",
        "recognition",
        "restoration",
        "parsing"
    ]
)

app.prepare(ctx_id=0, det_size=(640, 640))
img = cv2.imread("image.jpg")
assert img is not None
faces = app.get(img)
result = app.draw_on(img, faces)
assert cv2.imwrite("result.jpg", result)

Models

List

• codeformer.onnx codeformer2.onnx
• beauty GAN

Download

wget https://github.com/facefusion/facefusion-assets/releases/download/models/codeformer.onnx
wget https://github.com/facefusion/facefusion-assets/releases/download/models/face_parser.onnx
wget https://storage.googleapis.com/ailia-models/beauty_gan/G_ep300.onnx -O beautygan.onnx

Upload

pip install --upgrade huggingface_hub
huggingface-cli login
huggingface-cli upload maze/faceX det_10g.onnx det_10g.onnx
huggingface-cli upload maze/faceX 1k3d68.onnx 1k3d68.onnx
huggingface-cli upload maze/faceX genderage.onnx genderage.onnx
huggingface-cli upload maze/faceX w600k_r50.onnx w600k_r50.onnx

Reference

• insightface
• facer
• facefusion
• ailia
• sithu31296/EasyFace
• xinntao/facexlib