我正在尝试kotlin的PyTorch移动教程。我想在资产文件中加载模块"model.pt“。但不知道如何在资产文件中加载模块。
Java (用PyTorch移动教程"hello world“编写)
Module module = Module.load(assetFilePath(this, "model.pt"));kotlin
val module = Module.load("?????")发布于 2020-01-04 07:18:42
宣布这一职能:
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"))https://stackoverflow.com/questions/59588556
复制相似问题