YoloV5Face
This module implements pre- and post-processing for the YoloV5 model.
Supported Hardware (hw
):
cpu
igpu
gpu
Supported Models (model
):
yolov5n-face
(cpu
andigpu
only)yolov5s-face
Examples:
REST API:
WIP
Configuration File:
```yaml
source:
- address: videos/sd_london_station_sd.mp4
type: file
elements:
- name: infer
args:
model: yolov5s-face
hw: gpu
score_threshold: 0.3
iou_threshold: 0.3
```
References:
YoloV5Face Objects
class YoloV5Face(Detector)
YoloV5Face pre-processing and post-processing class.
Attributes:
models
list - List of supported models.