The following parameters are used in the Image Classification node.
Name: Input the node name used in a specific flow.
- default: image-classification
- type: string
Device: Select the target device mode to be used. Currently available are GPU, CPU, TPU or VPU.
- default: CPU
- type: string
In case you select VPU as your target device, you will be displayed with an additional field where you can indicate how many VPU target devices you would like to run your models on, or run the edge inference on both CPU and VPU. To do that, simply enter MYRIAD,MYRIAD,CPU as an example. This means, that the inference will be performed on 2x VPUs and 1x CPU.
Model Name: Select one of the most popular public models for image classification.
- Available models: See here
- type: string
Custom Model: You can link your own custom model by adding the model URL.
Detection Labels: Select the target object(s) to classified. A single or multiple objects can be selected.
- default: none
- type: string
Detection Score Threshold: Refers to the threshold value for confidence score filtering.
- default: 0.5
- range: (0.0~1.0)