Wrapper over YoloV5Face

Installation

pip install -U yolo5face

Inference

from yolo5face.get_model import get_model

model = get_model("yolov5n", gpu=-1, target_size=512, min_face=24)

image = cv2.imread(<IMAGE_PATH>)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

boxes, key_points = model(image)

gpu - GPU number, -1 or cpu for CPU
target_size - min size of the target_image
min_face - minimum face size in pixels. All faces that have side smaller than min_face will be ignored.

Download files

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

Source Distribution

yolo5face-0.0.4.tar.gz (17.0 kB view details)

Uploaded Jun 27, 2022 Source

File details

Details for the file yolo5face-0.0.4.tar.gz.

File metadata

Download URL: yolo5face-0.0.4.tar.gz
Upload date: Jun 27, 2022
Size: 17.0 kB
Tags: Source
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.23.0 setuptools/50.3.0.post20201006 requests-toolbelt/0.9.1 tqdm/4.50.2 CPython/3.7.3

File hashes

Hashes for yolo5face-0.0.4.tar.gz
Algorithm	Hash digest
SHA256	`44571948267ea3fbf1aa19477fd17a8eb2748bb6b807fe6c71a431bdad5db826`
MD5	`97ac11e5c2864cebbe72c240e174e649`
BLAKE2b-256	`072a01a9e402e45ea888a57385d4a90942f243e5ad27d23255f1a0217863bece`