Android(kotlin)，如何获取 Assets 文件路径？ (在 pytorch 手机上)

Question

我正在尝试 kotlin 的 PyTorch Mobile 教程。我想在 Assets 文件中加载模块“model.pt”。但不知道在 Assets 文件中加载模块。

java (写在 PyTorch 移动教程“hello world”中)

Module module = Module.load(assetFilePath(this, "model.pt"));

Kotlin

val module = Module.load("?????")

Answer 1

声明这个函数:

fun assetFilePath(context: Context, asset: String): String {
    val file = File(context.filesDir, asset)

    try {
        val inpStream: InputStream = context.assets.open(asset)
        try {
            val outStream = FileOutputStream(file, false)
            val buffer = ByteArray(4 * 1024)
            var read: Int

            while (true) {
                read = inpStream.read(buffer)
                if (read == -1) {
                    break
                }
                outStream.write(buffer, 0, read)
            }
            outStream.flush()
        } catch (ex: Exception) {
            e.printStackTrace()
        }
        return file.absolutePath
    } catch (e: Exception) {
        e.printStackTrace()
    }
    return ""
}

然后将其用作:

val module = Module.load(assetFilePath(this, "model.pt"))