Skip to main content

YoloV5

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

Supported Hardware (hw):

  • cpu
  • igpu

Supported Models (model):

  • yolov5s
  • yolov5s6

Examples:

REST API:

WIP

Configuration File:

```yaml
source:
- address: videos/sd_london_station_sd.mp4
  type: file
elements:
- name: infer
  args:
    model: yolov5s
    hw: cpu
    score_threshold: 0.3
    iou_threshold: 0.3
```

References:

YoloV5 Objects​

class YoloV5(Detector)

YoloV5 pre-processing and post-processing class.

Attributes:

  • models list - List of supported models.