基于Android pytorch API 深度学习图像识别（上）

目录

一、官方demo

1.1 demo 下载

1.2 demo 演示

 1.3 demo 介绍

1.4 TorchScript 模型

二、自己训练数据集

2.1 数据集

2.2 网络结构调整

2.3 训练代码 train.py

2.4 模型预测 predict.py

2.5 模型评估 eval.py

2.6 模型转换 jitmodel.py

2.7 测试转换后的模型 test_jitmodel.py

三、资源下载

---

一、官方demo

1.1 demo 下载

android-demo-app下载地址：git clone https://github.com/pytorch/android-demo-app.git

百度网盘链接：https://pan.baidu.com/s/1dnI6_fcYC71C6aJI2wrSIw  提取码：yavb

1.2 demo 演示

Android IDE：Android Studio 4.1.2

使用android打开HelloWorldApp工程：

查看readme文件，里面有详细介绍demo运行方法。

因本机IDE版本与官方不一样，加载项目出现错误：

修改原来gradle与gradle plugin 的版本。

在工程目录下的build.gradle中修改gradle plugin

dependencies {
    classpath 'com.android.tools.build:gradle:3.5.0'
}

为：

 dependencies {
        classpath 'com.android.tools.build:gradle:4.1.2'
    }

修改gradle版本为6.5