如何将pytorch模型部署到安卓(1)

load模型。这里要注意，如果保存模型

torch.save(model,‘models.pth’)

加载模型则是

model=torch.load(‘models.pth’)

如果保存模型是

torch.save(model.state_dict(),“models.pth”)

加载模型则是

model.load_state_dict(torch.load(‘models.pth’))

定义输入数据格式。

模型转化，然后再保存模型。

安卓部署

===============================================================

新建项目

---

新建安卓项目，选择Empy Activity,然后选择Next

然后，填写项目信息，选择安卓版本，我用的4.4，点击完成

导入包

---

导入pytorch_android的包

//pytorch

implementation ‘org.pytorch:pytorch_android:1.10.0’

implementation ‘org.pytorch:pytorch_android_torchvision:1.10.0’

如果有参数报错请参照我的完整的配置，代码如下：

plugins {

id ‘com.android.application’

}

android {

compileSdk 32

defaultConfig {

applicationId “com.example.myapplication”

minSdk 21

targetSdk 32

versionCode 1

versionName “1.0”

testInstrumentationRunner “androidx.test.runner.AndroidJUnitRunner”

}

buildTypes {

release {

minifyEnabled false

proguardFiles getDefaultProguardFile(‘proguard-android-optimize.txt’), ‘proguard-rules.pro’

}

}

<