helloworldapp项目
项目link:github.com/pytorch/and…
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);