使用 NV21 到位图的 Script Intrinsic YuvTo RGB（支持库）时图像损坏

Question

我在使用支持库中的 ScriptIntrinsicYuvToRGB 将图像从 NV21 格式转换为位图时遇到问题 (ARGB_8888)。下面的代码可以说明问题。

假设我有以下 50x50 图像（下面是设备的屏幕截图，实际上不是 50x50）：

然后如果我通过 YuvImage#compressToJpeg + BitmapFactory.decodeByteArray:

将所述图像转换为位图

YuvImage yuvImage = new YuvImage(example, android.graphics.ImageFormat.NV21, width, height, null);
ByteArrayOutputStream os = new ByteArrayOutputStream();
yuvImage.compressToJpeg(new Rect(0, 0, width, height), 100, os);
byte[] jpegByteArray = os.toByteArray();
return BitmapFactory.decodeByteArray(jpegByteArray, 0, jpegByteArray.length);

我得到了预期的图像。但是如果我通过 ScriptIntrinsicYuvToRGB 如下转换它：

RenderScript rs = RenderScript.create(context);

Type.Builder tb = new Type.Builder(rs, Element.createPixel(rs,
        Element.DataType.UNSIGNED_8, Element.DataKind.PIXEL_YUV));
tb.setX(width);
tb.setY(height);
tb.setYuvFormat(android.graphics.ImageFormat.NV21);
Allocation yuvAllocation = Allocation.createTyped(rs, tb.create(), Allocation.USAGE_SCRIPT);
yuvAllocation.copyFrom(example);

Type rgbType = Type.createXY(rs, Element.RGBA_8888(rs), width, height);
Allocation rgbAllocation = Allocation.createTyped(rs, rgbType);

ScriptIntrinsicYuvToRGB yuvToRgbScript = ScriptIntrinsicYuvToRGB.create(rs, Element.RGBA_8888(rs));
yuvToRgbScript.setInput(yuvAllocation);
yuvToRgbScript.forEach(rgbAllocation);

Bitmap convertedBitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
rgbAllocation.copyTo(convertedBitmap);

我得到以下损坏的图像：

我注意到这种情况发生在正方形尺寸的图像上，而不是 2 的幂（例如 64x64、128x128 等）。如果我没有尝试正方形尺寸，我就不会注意到这个问题，因为某些图片尺寸（例如 2592x1728）可以正常工作。我错过了什么？

更新：按要求放上生成原图的代码：

int width = 50;
int height = 50;
int size = width * height;

byte[] example = new byte[size + size / 2];
for (int x = 0; x < width; x++) {
    for (int y = 0; y < height; y++) {
        example[y * width + x] = (byte) ((x*y / (float)size) * 255);
    }
}
for (int i = 0; i < size / 2; i++) {
    example[size + i] = (byte) (127);
}

Answer 1

以下代码的行为方式有误：

Type.Builder tb = new Type.Builder(rs, Element.createPixel(rs,
                    Element.DataType.UNSIGNED_8, Element.DataKind.PIXEL_YUV));
tb.setX(width);
tb.setY(height);
tb.setYuvFormat(android.graphics.ImageFormat.NV21);
yuvAllocation = Allocation.createTyped(rs, tb.create(), Allocation.USAGE_SCRIPT);

如果您使用 "raw" 创建分配的方式替换它，转换将起作用：

int expectedBytes = width * height *
                ImageFormat.getBitsPerPixel(ImageFormat.NV21) / 8;

Type.Builder yuvTypeBuilder = new Type.Builder(rs, Element.U8(rs))
                                                   .setX(expectedBytes);
Type yuvType = yuvTypeBuilder.create();
yuvAllocation = Allocation.createTyped(rs, yuvType, Allocation.USAGE_SCRIPT);

看来，如果使用PIXEL_YUV定义，在非16的倍数维度上存在大小问题。仍在调查中。

Answer 2

恕我直言，在 Android 上将 NV21（或 NV12）转换为 ARGB 的最佳方法是使用本机 (C++) LibYUV 库 https://chromium.googlesource.com/libyuv/libyuv/

基于 ARM v7（和最新 v8）的 Android 设备的主要优势是 NEON 优化，这使得转换速度极快。

您可以自己构建 LibYUV (Build Instruction) or use any prebuild from github: https://github.com/search?q=libyuv+Android

使用 NV21 到位图的 Script Intrinsic YuvTo RGB（支持库）时图像损坏

Corrupted images when using ScriptIntrinsicYuvToRGB (support lib) for NV21 to Bitmap

android

bitmap

renderscript

yuv