TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mob][onnx] inline const

Browse Source
This commit is contained in:
Neeraj Gupta
2024-07-10 13:04:10 +05:30
parent 234efb63b2
commit 6b78ce715f
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mobile/plugins/onnx_dart/android/src/main/kotlin/io/ente/photos/onnx_dart/OnnxDartPlugin.kt
View File

@@ -39,9 +39,6 @@ class OnnxDartPlugin: FlutterPlugin, MethodCallHandler {
}
companion object {
const val DEFAULT_SESSION_COUNT = 1
const val K_INPUT_WIDTH = 640
const val K_INPUT_HEIGHT = 640
const val K_NUM_CHANNELS = 3
}
@@ -170,7 +167,7 @@ class OnnxDartPlugin: FlutterPlugin, MethodCallHandler {
try {
val env = OrtEnvironment.getEnvironment()
val inputTensorShape = longArrayOf(1, K_NUM_CHANNELS.toLong(), K_INPUT_HEIGHT.toLong(), K_INPUT_WIDTH.toLong())
val inputTensorShape = longArrayOf(1, 3, 640, 640)
val inputTensor = OnnxTensor.createTensor(env, FloatBuffer.wrap(inputData), inputTensorShape)
val inputs = mapOf("input" to inputTensor)
val outputs = session.run(inputs)