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

83 lines
1.5 KiB

3 years ago
# Face Embeddings using Deepface
*Author: Krishna katyal*
3 years ago
3 years ago
<br />
## Description
The pipeline is used to extract the feature vector of detected faces in images. It uses the for face embeddings [`Deepface`](https://github.com/serengil/deepface).
<br />
## Code Example
Load an image from path './test_face.jpg'.
3 years ago
*Write a pipeline with explicit inputs/outputs name specifications:*
```python
from towhee.dc2 import pipe, ops, DataCollection
3 years ago
p = (
pipe.input('path')
.map('path', 'img', ops.image_decode())
.map('img', 'vec', ops.face_embedding.deepface(model_name = 'DeepFace'))
.output('img', 'vec')
)
3 years ago
DataCollection(p('./test_face.jpg')).show()
```
3 years ago
<img src="./image.png" height="200px"/>
3 years ago
<br />
## Factory Constructor
Create the operator via the following factory method
***face_embedding.deepface(model_name = 'which model to use')***
Model options:
3 years ago
- VGG-Face
- FaceNet
- OpenFace
- DeepFace
- ArcFace
- Dlib
- DeepID
- Facenet512
3 years ago
<br />
## Interface
A face embedding operator takes a face image as input. It extracts the embedding in ndarray.
**Parameters:**
***img:*** *towhee.types.Image (a sub-class of numpy.ndarray)*
​ The input image.
**Returns:** *numpy.ndarray*
​ The extracted image embedding.
<br />
**Reference**
https://github.com/serengil/deepface
You can refer to [Getting Started with Towhee](https://towhee.io/) for more details. If you have any questions, you can [submit an issue to the towhee repository](https://github.com/towhee-io/towhee/issues).