/PyTorch_BlazeFace

Unofficial PyTorch implementation of BlazeFace

Primary LanguagePythonApache License 2.0Apache-2.0

PyTorch_BlazeFace

Build Status

Unofficial PyTorch implementation of BlazeFace

Installation

pip install git+https://github.com/tkat0/PyTorch_BlazeFace
import torch
from blazeface import BlazeFace

x = torch.randn(1, 3, 128, 128)
model = BlazeFace()
h = model(x)

Paper

BlazeFace: Sub-millisecond Neural Face Detection on Mobile GPUs

[Project Page] [Original Implementation]