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
mode: a
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.