Skip to main content

ImageClassification

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

Supported Hardware (hw):

  • cpu
  • igpu

Examples:

REST API:

WIP

Configuration File:

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

ImageClassification Objects

class ImageClassification(Classifier)

ImageClassification pre-processing and post-processing class.

Attributes:

  • models list - List of supported models.