pytorch Android部署 helloworldApp

helloworldapp项目

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

1.打开

pytorch Android部署 helloworldApp_第1张图片

2.在build.gradle中设置依赖

pytorch Android部署 helloworldApp_第2张图片

3.找到dependenciespytorch Android部署 helloworldApp_第3张图片

加入

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,然后点击锤子

pytorch Android部署 helloworldApp_第4张图片
可能会 报错
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);

5.点击三角,运行可以看到pytorch Android部署 helloworldApp_第5张图片

你可能感兴趣的:(深度学习,python,android,pytorch,模型部属)