TensorFlow Lite超分应用：ByteBuffer转Bitmap后无法显示求助

我来帮你分析下代码里的关键问题，正是这些问题导致Bitmap无法正常显示：

1. 错误使用BitmapFactory.decodeByteArray #

这个方法是用来解码JPEG/PNG等压缩格式的字节流的，而你的TensorFlow Lite输出是原始的像素数据（大概率是归一化后的浮点值），直接用它解析肯定得不到有效Bitmap，返回的是null，所以ImageView显示空白。

2. ByteBuffer的字节序未配置 #

TFLite默认使用native字节序（小端序），但你创建的out ByteBuffer没有指定字节序，这会导致读取到的像素数据完全错乱。

3. 像素值未做归一化逆转换 #

大多数图像超分辨率模型的输出是归一化到0-1或-1到1的浮点值，你需要把这些值转换回0-255的整数字节，才能生成正确的Bitmap。

修改后的完整代码 #

import java.nio.ByteOrder;

// 定义输出图像的参数
int outputWidth = 384;
int outputHeight = 384;
int channelCount = 3; // RGB通道

// 创建并配置ByteBuffer：用allocateDirect减少内存拷贝，指定native字节序
ByteBuffer out = ByteBuffer.allocateDirect(4 * outputWidth * outputHeight * channelCount);
out.order(ByteOrder.nativeOrder());

// 运行模型
tflite.run(byteBuffer, out);

// 重置ByteBuffer的读取位置到开头
out.rewind();

// 创建数组存储ARGB格式的像素（Bitmap默认用ARGB_8888）
int[] pixels = new int[outputWidth * outputHeight];

// 遍历每个像素，转换浮点值为RGB字节
for (int i = 0; i < outputWidth * outputHeight; i++) {
    // 读取模型输出的RGB浮点值（假设模型输出范围是0-1）
    float r = out.getFloat();
    float g = out.getFloat();
    float b = out.getFloat();

    // 将归一化值转换为0-255的整数，同时做边界限制避免溢出
    int red = Math.round(Math.max(0, Math.min(1, r)) * 255);
    int green = Math.round(Math.max(0, Math.min(1, g)) * 255);
    int blue = Math.round(Math.max(0, Math.min(1, b)) * 255);

    // 组合成ARGB格式（A通道设为255，表示完全不透明）
    pixels[i] = (255 << 24) | (red << 16) | (green << 8) | blue;
}

// 创建Bitmap并设置像素
Bitmap outPut_Image = Bitmap.createBitmap(outputWidth, outputHeight, Bitmap.Config.ARGB_8888);
outPut_Image.setPixels(pixels, 0, outputWidth, 0, 0, outputWidth, outputHeight);

// 显示到ImageView
ImageView imageView = findViewById(R.id.imageView2);
if (outPut_Image != null) {
    imageView.setImageBitmap(outPut_Image);
} else {
    Toast.makeText(this, "Bitmap创建失败", Toast.LENGTH_LONG).show();
}

额外注意事项 #

• 如果你的模型输出范围是-1到1，需要把转换逻辑改成：

  int red = Math.round(Math.max(0, Math.min(1, (r + 1) / 2)) * 255);
  

• 确认模型的输出通道顺序是RGB还是BGR，有些模型会输出BGR格式，这时需要调换red和blue的顺序。
• 可以通过Log.d打印outPut_Image的状态，确认是否成功创建：

  Log.d("BitmapStatus", outPut_Image == null ? "创建失败" : "尺寸：" + outPut_Image.getWidth() + "x" + outPut_Image.getHeight());
  

内容的提问来源于stack exchange，提问作者Syed Nauyan Rashid