Onnx runtime example in kotlin

onnx-example.kt

@Test
fun onnxRuntime() {
    val env = OrtEnvironment.getEnvironment()
    OrtSession.SessionOptions().use { opts ->
        opts.setOptimizationLevel(OrtSession.SessionOptions.OptLevel.BASIC_OPT)
        env.createSession("mymodel.onnx", opts).use { session ->
            for (i in session.inputInfo.values) {
                println(i)
            }
            for (o in session.outputInfo.values) {
                println(o)
            }
            val data = ByteBuffer.allocateDirect(1 * 3 * 64 * 64 * 4)
                    .order(ByteOrder.nativeOrder())
                    .asFloatBuffer()
            val tensor = OnnxTensor.createTensor(env, data, longArrayOf(1, 3, 64, 64))
            val output = session.run(Collections.singletonMap("images", tensor))
            println(output)
            val ot = output.get(0) as OnnxTensor
            val floatBuffer = ot.getDirectBuffer().asFloatBuffer()
            val toFloatArray = floatBuffer.toFloatArray()
            toFloatArray
        }
    }
}

gradle:

    implementation("com.microsoft.onnxruntime:onnxruntime:1.12.1")

on android

implementation("com.microsoft.onnxruntime:onnxruntime-android:1.12.1")