Package examples.onnx.cv

Functions

preprocessing

fun preprocessing(resizeTo: Pair<Int, Int>, model: OnnxInferenceModel): Operation<BufferedImage, Pair<FloatArray, TensorShape>>

runImageRecognitionPrediction

fun runImageRecognitionPrediction(modelType: ONNXModels.CV, executionProviders: List<ExecutionProvider> = emptyList()): List<Pair<String, Float>>

runONNXAdditionalTraining

fun runONNXAdditionalTraining(modelType: ONNXModels.CV, resizeTo: Pair<Int, Int> = Pair(224, 224))

This examples demonstrates the transfer learning concept on the Image Recognition model:

runONNXImageRecognitionPrediction

fun runONNXImageRecognitionPrediction(modelType: ONNXModels.CV, resizeTo: Pair<Int, Int> = Pair(224, 224))