Component implementation using YOLO.
To use this, please enable 'yolo' option under "Additional components" in the Face Editor section of the "Settings" tab.
This component utilizes an object detection model from YOLO for face detection. Though not exclusively designed for face detection, it can identify other objects of interest as well. Effectiveness can be enhanced by utilizing a model specifically trained on faces.
- YOLO
- N/A
path
(string, default: "yolov8n.pt"): Path to the model file. Ifrepo_id
is specified, the model will be downloaded from Hugging Face Model Hub instead, usingrepo_id
andfilename
.repo_id
(string, optional): The repository ID if the model is hosted on Hugging Face Model Hub. If this is specified,path
will be ignored.filename
(string, optional): The filename of the model in the Hugging Face Model Hub repository. Use this in combination withrepo_id
.conf
: (float, optional, default: 0.5): The confidence threshold for object detection.
- tag: The class of the detected object (as trained in the YOLO model)
- attributes: N/A
- landmarks: N/A
This component utilizes a segmentation model from YOLO (You Only Look Once) for mask generation.
- YOLO
- Use minimal area (for close faces)
path
(string, default: "yolov8n-seg.pt"): Path to the model file. Ifrepo_id
is specified, the model will be downloaded from Hugging Face Model Hub instead, usingrepo_id
andfilename
.repo_id
(string, optional): The repository ID if the model is hosted on Hugging Face Model Hub. If this is specified,path
will be ignored.filename
(string, optional): The filename of the model in the Hugging Face Model Hub repository. Use this in combination withrepo_id
.conf
(float, default: 0.5): Confidence threshold for detections. Any detection with a confidence lower than this will be ignored.