手把手搭建一个pytorch mobile 的Android APP

pytorch 在2019年快年底发布了pytorch mobile,用于移动端的部署,虽然比起TensorFlow lite晚了很久,终归还可以做一些事了。

pytorch mobile运行的远离是在手机端调用c++操作torchscript文件,因此我们要想能够运行训练好的模型,就必须转成torchscript文件。

转torchscript文件有专门的api,这里是一个例子:

import torch
import torchvision
resnet18 = torchvision.models.resnet18(pretrained=True)
resnet18.eval()
example_inputs = torch.rand(1, 3, 224, 224)
resnet18_traced = torch.jit.trace(resnet18, example_inputs = example_inputs)
resnet18_traced.save("resnet18_traced.pt")

我们可以看到调用了torch.jit.trace这个方法进行转换,转换成专门的文件后放到Android里面的assert文件夹下,我们就能读取这个模型

注意:上面的代码只是演示api,跟接下来跑这个程序无关,只是为了了解这个过程而已,另外我下面的链接其实已经包括了那个.pt文件,不需要自己去转换,这里只是要知道文件怎么来的

接下来运行的也是演示的官网的例子(5ehc)

剩下的过程和之前TensorFlow lite是一样的

  • 2
    点赞
  • 32
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
PyTorch Mobile一个用于在移动设备上部署和运行 PyTorch 模型的工具。它允许您将训练好的 PyTorch 模型转换为适用于移动设备的格式,并在移动应用程序中进行推理。 首先,您需要导入必要的库,并加载一个预训练的模型,例如 `torchvision.models.mobilenet_v2`。接下来,您可以准备一个输入样例,并使用 `torch.jit.trace` 方法将模型转换为 TorchScript 格式。然后,您可以使用 `torch.utils.mobile_optimizer.optimize_for_mobile` 方法对 TorchScript 模型进行优化,以减小模型的大小并提高推理速度。最后,使用 `traced_script_module_optimized._save_for_lite_interpreter` 方法将优化后的模型保存为 `.ptl` 文件,以供移动应用程序使用。 在准备输入时,您可以使用 `org.pytorch.torchvision.TensorImageUtils` 子库中的 `bitmapToFloat32Tensor` 方法,将 Android 的位图转换为 PyTorch 可以接受的张量形式。这个方法可以帮助您将图像数据转换为适用于模型推理的输入格式。 <span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [Pytorch-Mobile-Android(2)](https://blog.csdn.net/unamable/article/details/118698349)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值