ONNX格式下图像分类模型输入标准化（java版）

大家在使用onnx格式的图像分类模型或者其他需要输入图像的模型时会遇到输入图像格式问题，这点onnx官方给出的案例模板中已经有现成的kotlin代码，但是萌新使用java进行安卓开发时不会使用kotlin，所以我写了一份java版的处理代码给大家参考

目录

一、图像处理类

二、使用方法

---

一、图像处理类

import android.graphics.Bitmap;

import java.nio.FloatBuffer;

public class ImagePreProcessor {

    private static final int DIM_BATCH_SIZE = 1;
    private static final int DIM_PIXEL_SIZE = 3;
    private static final int IMAGE_SIZE_X = 224;
    private static final int IMAGE_SIZE_Y = 224;

    public static FloatBuffer preProcess(Bitmap bitmap) {
        FloatBuffer imgData = FloatBuffer.allocate(
                DIM_BATCH_SIZE
                        * DIM_PIXEL_SIZE
                        * IMAGE_SIZE_X
                        * IMAGE_SIZE_Y
        );
        imgData.rewind();
        int stride = IMAGE_SIZE_X * IMAGE_SIZE_Y;
        int[] bmpData