体验PyTorch官方Helloworld

一、Android HelloWorldApp

1 准备工作

提前安装anaconda

下载源码

https://github.com/pytorch/android-demo-app

改用阿里源

maven { url 'https://maven.aliyun.com/repository/google' }
maven{ url 'https://maven.aliyun.com/repository/jcenter'}

改一下依赖,删掉SNAPSHOT

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

2 准备模型

安装工具

pip install torchvision

运行脚本,得到model.pt模型。

python trace_model.py

3 运行

编译运行APP。

识别出狼

换了小黄鸭的照片,识别出是奶嘴。

旗杆

熊猫

二、PyTorchDemoApp

这里面有几个示例:实时拍摄识别物体

输入一段文字自动评分。

参考

android studio解决pytorch-android-1.6.0版本arr下载慢的问题_构建的乐趣的博客-CSDN博客
https://blog.csdn.net/qq_44065334/article/details/113135490

Android | 火炬 helloworld的文档
https://pytorch.org/mobile/android/#quickstart-with-a-helloworld-example

Installing on macOS — Anaconda documentation
https://docs.anaconda.com/anaconda/install/mac-os/

android-demo-app/HelloWorldApp at master · pytorch/android-demo-app
https://github.com/pytorch/android-demo-app/tree/master/HelloWorldApp

posted @ 2021-05-29 18:14  奥巴荣  阅读(575)  评论(0编辑  收藏  举报