Skip to main content

YoloV5Face

This module implements pre- and post-processing for the YoloV5 model.

Supported Hardware (hw):

  • cpu
  • igpu
  • gpu

Supported Models (model):

  • yolov5n-face (cpu and igpu 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.