这个模型在安卓对应的包：
repositories {
jcenter()
}
dependencies {
implementation ‘org.pytorch:pytorch_android_lite:1.9.0’
implementation ‘:pytorch_android_torchvision:1.9.0’
}
注:pytorch_android_lite版本和转化模型用的版本要一致，不一致就会报各种错误。
目前用这种方法有点问题，我采用的另一种方法。
转化代码如下：
import torch
import
pytorch环境中
model_pth = ‘model_31_0.96.pth’ #模型的参数文件
mobile_pt =‘’ # 将模型保存为Android可以调用的文件
model = (model_pth)
() # 模型设为评估模式
device = (‘cpu’)
(device)
1张3通道224*224的图片
input_tensor = (1, 3, 224, 224) # 设定输入数据格式
mobile = (model, input_tensor) # 模型转化
(mobile_pt) # 保存文件
对应的包：
//pytorch
implementation ‘:pytorch_android:1.10.0’
implementation ‘:pytorch_android_torchvision:1.10.0’
定义模型文件和转化后的文件路径。
load模型。这里要注意，如果保存模型
(model,‘models.pth’)
加载模型则是
model=(‘’)
如果保存模型是
(model.state_dict(),“”)
加载模型则是
model.load_state_dict((‘’))
定义输入数据格式。
模型转化，然后再保存模型。
安卓部署
===============================================================
新建项目
新建安卓项目，选择Empy Activity,然后选择Next
然后，填写项目信息，选择安卓版本，我用的4.4，点击完成
导入包
导入pytorch_android的包
//pytorch
implementation ‘:pytorch_android:1.10.0’
implementation ‘:pytorch_android_torchvision:1.10.0’
如果有参数报错请参照我的完整的配置，代码如下：
plugins {
id ‘’
}
android {
compileSdk 32
defaultConfig {
applicationId “”
minSdk 21
targetSdk 32
versionCode 1
versionName “1.0”
testInstrumentationRunner “”
}
buildTypes {
release {
minifyEnabled false
proguardFiles getDefaultProguardFile(‘’), ‘’
}
}
compileOptions {
sourceCompatibility JavaVersion.VERSION_1_8
targetCompatibility JavaVersion.VERSION_1_8
}
}
dependencies {
implementation ‘:appcompat:1.3.0’
implementation ‘:material:1.4.0’
implementation ‘:constraintlayout:2.0.4’
testImplementation ‘junit:junit:4.13.2’
androidTestImplementation ‘:junit:1.1.3’
androidTestImplementation ‘:espresso-core:3.4.0’
//pytorch
implementation ‘:pytorch_android:1.10.0’
implementation ‘:pytorch_android_torchvision:1.10.0’
}
页面文件
页面的配置如下：
<?xml version="1.0" encoding="utf-8"?><FrameLayout xmlns:android=“/apk/res/android”
xmlns:tools=“/tools”
android:layout_width=“match_parent”
android:layout_height=“match_parent”
tools:context=“.MainActivity”>
<ImageView
android:id=“@+id/image”
android:layout_width=“match_parent”
android:layout_height=“match_parent”
android:scaleType=“fitCenter” />
<TextView
android:id=“@+id/text”
android:layout_width=“match_parent”
android:layout_height=“wrap_content”
android:layout_gravity=“top”
android:textSize=“24sp”
android:background=“#80000000”
android:textColor=“@android:color/holo_red_light” />
这个页面只有两个空间，一个展示图片，一个显示文字。
模型推理
新增assets文件夹，然后将转化的模型和待测试的图片放进去。
新增ImageNetClasses类，这个类存放类别名字。
代码如下：
package ;
public class ImageNetClasses {
public static String[] IMAGENET_CLASSES = new String[]{
“Black-grass”,
“Charlock”,
“Cleavers”,
“Common Chickweed”,
“Common wheat”,
“Fat Hen”,
“Loose Silky-bent”,
“Maize”,
“Scentless Mayweed”,
“Shepherds Purse”,
“Small-flowered Cranesbill”,
“Sugar beet”,
};
}
在MainActivity类中，增加模型推理的逻辑。完成代码如下：
自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。
深知大多数初中级Android工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则近万的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！
因此收集整理了一份《2024年Android移动开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。
既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Android开发知识点，真正体系化！
由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！
如果你觉得这些内容对你有帮助，可以扫码获取！！（备注：Android）
文末
那么对于想坚持程序员这行的真的就一点希望都没有吗？
其实不然，在互联网的大浪淘沙之下，留下的永远是最优秀的，我们考虑的不是哪个行业差哪个行业难，就逃避掉这些，无论哪个行业，都会有他的问题，但是无论哪个行业都会有站在最顶端的那群人。我们要做的就是努力提升自己，让自己站在最顶端，学历不够那就去读，知识不够那就去学。人之所以为人，不就是有解决问题的能力吗？挡住自己的由于只有自己。
Android希望=技能+面试
-
技能
-
面试技巧+面试题
《Android学习笔记总结+移动架构视频+大厂面试真题+项目实战源码》，点击传送门即可获取！
文末
那么对于想坚持程序员这行的真的就一点希望都没有吗？
其实不然，在互联网的大浪淘沙之下，留下的永远是最优秀的，我们考虑的不是哪个行业差哪个行业难，就逃避掉这些，无论哪个行业，都会有他的问题，但是无论哪个行业都会有站在最顶端的那群人。我们要做的就是努力提升自己，让自己站在最顶端，学历不够那就去读，知识不够那就去学。人之所以为人，不就是有解决问题的能力吗？挡住自己的由于只有自己。
Android希望=技能+面试
-
技能
[外链图片转存中…(img-jtSI0XpP-1712393659162)] -
面试技巧+面试题
[外链图片转存中…(img-pwBxNxyB-1712393659163)]