logo
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions

Updated 3 years ago

face-landmark-detection

MobileFaceNet Face Landmark Detecter

authors: David Wang

Desription

MobileFaceNets is a class of extremely efficient CNN models to extract 68 landmarks from a facial image, which use less than 1 million parameters and are specifically tailored for high-accuracy real-time face verification on mobile and embedded devices. This repo is an adaptation from cuijian/pytorch_face_landmark.

Code Example

Extract facial landmarks from './img1.jpg'.

Write the pipeline in simplified style:

import towhee

towhee.glob('./img1.jpg') \
  .image_decode.cv2() \
  .face_landmark_detection.mobilefacenet() \
  .select('img','landmark') \
  .to_list()

Write a same pipeline with explicit inputs/outputs name specifications:

import towhee

towhee.glob['path']('./img1.jpg') \
  .image_decode.cv2['path', 'img']() \
  .face_landmark_detection.mobilefacenet['img', 'landmark']() \
  .select('img','landmark') \
  .show()
result1

Factory Constructor

Create the operator via the following factory method

face_landmark_detection.mobilefacenet(pretrained = True)

Parameters:

pretrained

​ whether load the pretrained weights.

​ supported types: bool, default is True, using pretrained weights.

Interface

An image embedding operator takes an image as input. it extracts the embedding back to ndarray.

Parameters:

img: towhee.types.Image (a sub-class of numpy.ndarray)

​ The input image.

Returns: numpy.ndarray

​ The extracted facial landmarks.

wxywb 597e959f1c update the readme. 4 Commits
file-icon .gitattributes
1.1 KiB
download-icon
Initial commit 3 years ago
file-icon README.md
1.7 KiB
download-icon
update the readme. 3 years ago
file-icon __init__.py
711 B
download-icon
update the readme. 3 years ago
file-icon mobilefacenet.py
2.5 KiB
download-icon
update the readme. 3 years ago
file-icon mobilefacenet_impl.py
5.9 KiB
download-icon
update the operator. 3 years ago
file-icon mobilefacenet_model_best.pth
12 MiB
download-icon
update the operator. 3 years ago
file-icon result.png
137 KiB
download-icon
update the readme. 3 years ago