基于Tensorflow Lite的Android Java人脸表情识别

dependencies {
    implementation 'org.tensorflow:tensorflow-lite:2.4.1' // 使用最新版本
}

import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.os.Bundle;
import android.util.Log;
import androidx.appcompat.app.AppCompatActivity;
import org.tensorflow.lite.Interpreter;
import java.io.FileInputStream;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.nio.MappedByteBuffer;
import java.nio.channels.FileChannel;

public class MainActivity extends AppCompatActivity {

    private Interpreter tflite;
    private MappedByteBuffer tfliteModel;
    private Bitmap inputBitmap;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        // 加载模型
        try {
            tfliteModel = FileUtil.loadMappedFile(this, "model.tflite");
            tflite = new Interpreter(tfliteModel);
        } catch (IOException e) {
            Log.e("MainActivity", "Error loading model", e);
        }

        // 加载并预处理输入图像
        inputBitmap = BitmapFactory.decodeResource(getResources(), R.drawable.input_image);
        Bitmap resizedBitmap = Bitmap.createScaledBitmap(inputBitmap, 48, 48, true); // 根据模型输入尺寸调整
        float[] input = preprocessImage(resizedBitmap);

        // 创建输出数组
        float[][] output = new float[1][7]; // 假设模型输出7种表情的概率

        // 运行推理
        tflite.run(input, output);

        // 处理输出结果
        int predictedClass = argMax(output[0]);
        Log.d("MainActivity", "Predicted expression: " + predictedClass);
    }

    private float[] preprocessImage(Bitmap bitmap) {
        int width = bitmap.getWidth();
        int height = bitmap.getHeight();
        float[] floatValues = new float[width * height * 3];
        int[] intValues = new int[width * height];
        bitmap.getPixels(intValues, 0, width, 0, 0, width, height);
        for (int i = 0; i < intValues.length; i++) {
            final int val = intValues[i];
            floatValues[i * 3] = ((val >> 16) & 0xFF) / 255.0f;
            floatValues[i * 3 + 1] = ((val >> 8) & 0xFF) / 255.0f;
            floatValues[i * 3 + 2] = (val & 0xFF) / 255.0f;
        }
        return floatValues;
    }

    private int argMax(float[] array) {
        int maxIndex = 0;
        float maxValue = array[0];
        for (int i = 1; i < array.length; i++) {
            if (array[i] > maxValue) {
                maxValue = array[i];
                maxIndex = i;
            }
        }
        return maxIndex;
    }
}

<uses-permission android:name="android.permission.READ_EXTERNAL_STORAGE"/>