torch.jit.ScriptModule导出的ptl文件在Android加载.getDataAsFloatArray出错
环境:
Implementation 'org.pytorch:pytorch_android_lite:1.10.0'
Implementation 'org.pytorch:pytorch_android_torchvision:1.9.0'
代码:
【模型加载处】
Publicvoid loadModule(AppCompatActivityapp) {
try {
module = LiteModuleLoader.load(assetFilePath(app, model_fname+".ptl"));
} catch (IOExceptione) {
Log.e("Pytorch", "Errorreadingassets", e);
app.finish();
}
}
【模型使用处】
//running the model
final Tensor outputTensor = module.forward(IValue.from(inputTensor)).toTensor();
//getting tensor content as java array of floats
finalfloat [] scores = outputTensor.getDataAsFloatArray();
错误描述:
模型加载时未出现任何错误,但使用处.getDataAsFloatArray提示输出为IntegerArray类型,且结果恒为[0]
原因及修订方式:
ptl文件名不允许有下划线("_"),加载的模型名称为"test_model.ptl",修改为"testNC.ptl"即恢复正常
A geek and poetry lover.
分类:
Android的神经网络开发
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本
· C#/.NET/.NET Core优秀项目和框架2025年2月简报
· Manus爆火,是硬核还是营销?
· 终于写完轮子一部分:tcp代理 了,记录一下
· 【杭电多校比赛记录】2025“钉耙编程”中国大学生算法设计春季联赛(1)