pytorch Android部署 helloworldApp

helloworldapp项目项目link：https://github.com/pytorch/android-demo-app1.打开2.在build.gradle中设置依赖3.找到dependencies加入implementation'org.pytorch:pytorch_android:1.8.0'implementation'org.pytorch:pytorch_android_t

RichardorMu

744人浏览 · 2021-07-16 11:19:19

RichardorMu · 2021-07-16 11:19:19 发布

helloworldapp项目

项目link：https://github.com/pytorch/android-demo-app

1.打开

在这里插入图片描述

2.在build.gradle中设置依赖

在这里插入图片描述

3.找到dependencies

加入

implementation'org.pytorch:pytorch_android:1.8.0'
implementation'org.pytorch:pytorch_android_torchvision:1.8.0'

请注意，如果你按照官网github给的例子https://github.com/pytorch/android-demo-app/blob/367d2d9448d453721f33f4cd6d1911627647016f/HelloWorldApp/app/build.gradle#L28-L29
可能会因为网络原因出错

implementation 'org.pytorch:pytorch_android:1.8.0-SNAPSHOT'
implementation 'org.pytorch:pytorch_android_torchvision:1.8.0-SNAPSHOT'

因此将snapshot去掉

4.随后右上角选择MainActivity，然后点击锤子

在这里插入图片描述
可能会报错
HelloWorldApp - Build: failed - error: no suitable method found for bitmapToFloat32Tensor(Bitmap,float[],float[],MemoryFormat)
为51行的将图片转换格式
将其中的MemoryFormat.CHANNELS_LAST删除即可

final Tensor inputTensor = TensorImageUtils.bitmapToFloat32Tensor(bitmap,TensorImageUtils.TORCHVISION_NORM_MEAN_RGB, TensorImageUtils.TORCHVISION_NORM_STD_RGB, MemoryFormat.CHANNELS_LAST);