Android开发：如何获取ONNX文件路径以创建OrtSession？

一、ONNX文件的正确存放方案 #

• 优先选：放入assets目录
  assets目录的文件无法直接获取绝对路径，但可以先将其复制到应用私有存储的files目录，再用该目录下的文件路径创建会话，完美避开内存溢出问题。
• 不推荐：放入res/raw或普通res目录
  这类目录的文件需要读取到内存才能使用，大文件依然会触发OOM，和你之前遇到的问题一致。

二、具体实现步骤 #

1. 编写工具方法：复制assets文件到私有存储 #

这个方法会先检查文件是否已存在，避免重复复制：

private fun copyAssetToFilesDir(context: Context, assetFileName: String): String {
    val destFile = File(context.filesDir, assetFileName)
    // 文件已存在直接返回路径
    if (destFile.exists()) {
        return destFile.absolutePath
    }
    // 复制assets文件到目标路径
    context.assets.open(assetFileName).use { inputStream ->
        FileOutputStream(destFile).use { outputStream ->
            inputStream.copyTo(outputStream)
        }
    }
    return destFile.absolutePath
}

2. 使用路径创建ONNX会话 #

在你的代码中调用上述方法获取路径，再传入createSession：

val sessionOptions: OrtSession.SessionOptions = OrtSession.SessionOptions()
sessionOptions.registerCustomOpLibrary(OrtxPackage.getLibraryPath())
// 获取复制后的ONNX文件绝对路径
val onnxFilePath = copyAssetToFilesDir(applicationContext, "l1_decoder.onnx")
// 直接用路径创建会话，无需加载文件到内存
ortSession = ortEnv.createSession(onnxFilePath, sessionOptions)

三、注意事项 #

• 复制操作建议放在子线程执行，避免阻塞主线程导致ANR。
• 应用私有存储的文件仅当前应用可访问，无需担心文件泄露。
• 如果ONNX文件后续需要更新，可以在复制前判断文件版本，再决定是否覆盖旧文件。

内容的提问来源于stack exchange，提问作者suihe dolood