Core ML相关问题：由RGBA字节数组生成的UIImage无法完整显示

嘿，我来帮你捋捋这个Core ML场景下的图片转换问题，你的代码里有几个容易忽略的细节，正是这些导致图片显示不全。咱们一步步来修复：

核心问题排查与修复 #

1. Alpha通道格式不匹配 #

你代码里用了kCGImageAlphaPremultipliedLast，这个参数要求你的RGBA字节数组是预乘Alpha格式（也就是RGB通道的值已经和Alpha值相乘过，比如RGB值=原始RGB*Alpha/255）。但Core ML输出的张量通常是非预乘的RGBA格式（每个通道独立0-255），这时候用预乘参数会导致颜色错位、显示区域异常。

👉 修复：把kCGImageAlphaPremultipliedLast改成kCGImageAlphaLast，如果确认原始数据是预乘格式再改回去。

2. UIImage的Scale参数设置 #

你用了scale:0，虽然iOS会自动适配，但在Core ML的特定尺寸输出场景下，手动指定屏幕scale更稳妥，避免图片被错误拉伸或裁剪。

👉 修复：把scale:0换成scale:[UIScreen mainScreen].scale。

3. 字节数组长度与内存泄漏检查 #

• 先确认你的bytes总长度是width * height * 4，如果长度不够，图片必然显示不全；
• 别忘了释放CGImageRef，你原来的代码里没写CGImageRelease(cgImage)，这会造成内存泄漏。

修复后的完整代码 #

// 假设你的RGBA字节数组是非预乘Alpha格式
CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
CGBitmapInfo bitmapInfo = kCGImageAlphaLast; // 非预乘Alpha用这个，预乘用kCGImageAlphaPremultipliedLast

CGContextRef context = CGBitmapContextCreate(
    bytes,
    width,
    height,
    8, // 每个通道8位，符合RGBA8888格式
    4 * width, // 每行字节数，RGBA每个像素4字节，确保字节对齐
    colorSpace,
    bitmapInfo
);

if (!context) {
    NSLog(@"创建Bitmap上下文失败！");
    CFRelease(colorSpace);
    return nil;
}

CFRelease(colorSpace);
CGImageRef cgImage = CGBitmapContextCreateImage(context);
CGContextRelease(context);

// 指定屏幕scale，保证显示尺寸正确
UIImage *image = [UIImage imageWithCGImage:cgImage
                                     scale:[UIScreen mainScreen].scale
                               orientation:UIImageOrientationUp];

CGImageRelease(cgImage); // 释放CGImage资源

额外排查点 #

如果还是有问题，再检查这两个地方：

• Core ML输出的张量尺寸：确认width和height和模型输出的H、W完全一致，比如有些模型输出是NHWC格式，要先提取正确的高度和宽度；
• 数据类型转换：如果Core ML输出的是float32类型的张量，要先把每个通道的值映射到0-255的UInt8范围再转成字节数组，直接强转会导致数据异常。

内容的提问来源于stack exchange，提问作者gi097