Package examples.onnx.cv

Functions

preprocessing

fun preprocessing(resizeTo: Pair<Int, Int>, model: OnnxInferenceModel): Preprocessing

runImageRecognitionPrediction

fun runImageRecognitionPrediction(modelType: ONNXModels.CV<OnnxInferenceModel>)

runONNXAdditionalTraining

fun runONNXAdditionalTraining(modelType: ONNXModels.CV<out OnnxInferenceModel>, 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<out OnnxInferenceModel>, resizeTo: Pair<Int, Int> = Pair(224, 224))