Skip to content

Image Classification

Purpose of the Node

The Image Classification node predicts the class of an input image using an ONNX model. It applies default transformations to the input image such as resizing, center cropping, normalizing, and scaling pixel values to ensure compatibility with the ONNX model. The node also supports optional softmax activation to scale the output logits to probabilities.

Pins

Pin NamePin DescriptionPin TypeValue Type
StartInitiate ExecutionExecution-
ModelONNX Model SessionStructNodeOnnxSession
ImageImage ObjectStructNodeImage
MeanImage Mean for Normalization (per channel)FloatArray
StdImage Standard Deviation for Normalization (per channel)FloatArray
CropCenter Crop PercentageFloat-
Softmax?Scale Outputs with SoftmaxBoolean-
EndDone with the ExecutionExecution-
PredictionsClass PredictionsStructArray