Wells0 2022-04-25 16:48 采纳率: 80%
浏览 57
已结题

关于安卓部署AI,现在手上只有一个torchscript

模型是用来人像分割的,模型源码也是有的,但不是本人负责,如果能通过这个torchscript进行部署是比较方便的。
部署效果是输入一个视频,经过模型处理后输出一个视频。
(如果能附上学习链接就更好了,有点不知道从何下手

  • 写回答

1条回答 默认 最新

  • Wells0 2022-04-30 16:19
    关注

    自问自答,目前进度是能使用模型处理图像。
    首先是关于模型:torchscript模型是可以通过pytorch.jit.load直接进行导入,可以通过netron.app查看模型的结构。Android提供了两种库 pytorch_android和pytorch_android_lite(版本需要与导出模型文件的python torch版本相同),这边使用了后者,对应于安卓中torchModuleLoader.load(),在python导出模型需要用_save_for_lite_interpreter的方式(详细导出可见pytorch官网)。于是模型就能载入了。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 5月8日
  • 已采纳回答 4月30日
  • 创建了问题 4月25日