开源项目 `tflite-android-transformers` 使用教程

开源项目 tflite-android-transformers 使用教程

tflite-android-transformersDistilBERT / GPT-2 for on-device inference thanks to TensorFlow Lite with Android demo apps项目地址:https://gitcode.com/gh_mirrors/tf/tflite-android-transformers

1. 项目的目录结构及介绍

tflite-android-transformers/
├── app/
│   ├── build.gradle
│   ├── proguard-rules.pro
│   └── src/
│       ├── main/
│       │   ├── AndroidManifest.xml
│       │   ├── java/
│       │   │   └── com/
│       │   │       └── huggingface/
│       │   │           └── transformers/
│       │   │               ├── MainActivity.java
│       │   │               └── ...
│       │   └── res/
│       │       ├── drawable/
│       │       ├── layout/
│       │       ├── mipmap/
│       │       └── values/
│       └── test/
│           └── ...
├── build.gradle
├── gradle.properties
├── gradlew
├── gradlew.bat
├── settings.gradle
└── ...

目录结构介绍

• app/: 包含应用程序的主要代码和资源。
  • build.gradle: 应用程序的构建脚本。
  • proguard-rules.pro: ProGuard 规则文件。
  • src/: 源代码和资源文件。
    • main/: 主源代码和资源。
      • AndroidManifest.xml: 应用程序的清单文件。
      • java/: Java 源代码。
        • com/huggingface/transformers/: 主要逻辑代码。
          • MainActivity.java: 应用程序的主活动。
      • res/: 资源文件，包括布局、图片和字符串等。
    • test/: 测试代码。
• build.gradle: 项目的根构建脚本。
• gradle.properties: Gradle 属性配置。
• gradlew 和 gradlew.bat: Gradle 包装脚本。
• settings.gradle: 项目设置文件。

2. 项目的启动文件介绍

MainActivity.java

MainActivity.java 是应用程序的入口点，负责初始化界面和处理用户交互。以下是该文件的主要内容：

package com.huggingface.transformers;

import android.os.Bundle;
import androidx.appcompat.app.AppCompatActivity;

public class MainActivity extends AppCompatActivity {
    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        // 初始化逻辑
    }
}

功能介绍

• onCreate(Bundle savedInstanceState): 当活动被创建时调用，主要负责设置布局和初始化组件。

3. 项目的配置文件介绍

build.gradle

build.gradle 文件位于项目根目录和 app/ 目录下，分别用于配置整个项目和应用程序模块。

根目录下的 build.gradle

// 根目录下的 build.gradle
buildscript {
    repositories {
        google()
        mavenCentral()
    }
    dependencies {
        classpath 'com.android.tools.build:gradle:7.0.0'
    }
}

allprojects {
    repositories {
        google()
        mavenCentral()
    }
}

app/ 目录下的 build.gradle

// app/ 目录下的 build.gradle
apply plugin: 'com.android.application'

android {
    compileSdkVersion 30
    defaultConfig {
        applicationId "com.huggingface.transformers"
        minSdkVersion 21
        targetSdkVersion 30
        versionCode 1
        versionName "1.0"
    }
    buildTypes {
        release {
            minifyEnabled false
            proguardFiles getDefaultProguardFile('proguard-android-optimize.txt'), 'proguard-rules.pro'
        }
    }
}

dependencies {
    implementation 'androidx.appcompat:appcompat:1.3.1'
    implementation 'com.google.android.material:material:1.4.0'
    implementation 'org.tensorflow:tensorflow-lite:2.5.0'
}

功能介绍

• buildscript: 配置构建

tflite-android-transformersDistilBERT / GPT-2 for on-device inference thanks to TensorFlow Lite with Android demo apps项目地址:https://gitcode.com/gh_mirrors/tf/tflite-android-transformers