Skip to main content

Packaged version YOLOv5 Face Detector

Project description

Yolov5 Face Detection


The project is a wrap over yolov5-face repo. Made simple portable interface for model import and inference. Model detects faces on images and returns bounding boxes and coordinates of 5 facial keypoints, which can be used for face alignment.


pip install yolov5facedetector

Usage example

from yolov5facedetector.face_detector import YoloDetector
import numpy as np
from PIL import Image

model = YoloDetector(target_size=720,gpu=0,min_face=90)
rgb_array_img = np.array('test_image.jpg')) # Will make RGB Numpy Array Image
bboxes, confs, points = model.predict(rgb_array_img)

You can also pass several images packed in a list to get multi-image predictions.

bboxes, confs, points = model.predict([image1,image2])

Other pretrained models

Currently Support YOLOv5 type n,m & l type of model from yolov5-face repo. Default model type is 'yolov5n' but you can change to m & l (larger model version) which is auto download when first time used

Example below:

model = YoloFace(yolo_type='yolov5l',target_size=720) # Will download weight file automatically
bboxes, confs, points = model.predict(rgb_array_img)

Result example


Thanks to deepcam-cn for pretrained models and Rebrikov Artem for providing wrapper function of YOLOv5Face.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

yolov5facedetector-1.0.2.tar.gz (164.7 kB view hashes)

Uploaded source

Built Distribution

yolov5facedetector-1.0.2-py3-none-any.whl (102.3 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page