ImageRecognitionModel

fun ImageRecognitionModel(internalModel: OnnxInferenceModel, channelsFirst: Boolean, preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>> = Identity(), modelKindDescription: String? = null, classLabels: Map<Int, String> = Imagenet.V1k.labels())

Content copied to clipboard