EfficientNetB2

class EfficientNetB2(noTop: Boolean) : ONNXModels.CV<OnnxInferenceModel>

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.api.core.util.loadImageNetClassLabels method).

Instantiates the EfficientNetB2 architecture.

The model have

an input with the shape (1x260x260x3)
an output with the shape (1x1000)
an output for noTop model with the shape (1x8x8x1408)

NOTE: This model is converted from Keras.applications, the last two layers in the noTop model have been removed so that the user can fine-tune the model for his specific task.