Wrapper over Yolo5Face for a more convenient inference.
Project description
Wrapper over YoloV5Face
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
orcpu
for CPUtarget_size
- min size of the target_imagemin_face
- minimum face size in pixels. All faces that have side smaller thanmin_face
will be ignored.
Project details
Release history Release notifications | RSS feed
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.3.tar.gz
(17.5 kB
view hashes)