如何在UIImage中导出RGB数据？获取UInt8数组及RGB与RGBA差异咨询

嘿，我来帮你理清楚这两个问题：

一、RGB 和 RGBA 的核心区别 #

简单来说就是通道数量和用途不同：

• RGB：是三通道颜色模式，每个像素由红(R)、绿(G)、蓝(B)三个分量组成，每个分量取值0-255，总共占3个字节。它只负责表示颜色本身，没有透明度信息。
• RGBA：在RGB基础上多了一个Alpha(A)通道，这个通道用来控制像素的透明度——0代表完全透明，255代表完全不透明，每个像素占4个字节。iOS的很多图形框架（比如UIKit、Core Graphics）默认用RGBA，是因为日常开发中经常需要处理视图叠加、半透明效果这类场景，带Alpha通道会更灵活。

二、提取纯RGB字节数组的方法（Swift/Objective-C） #

如果你只需要RGB数据（无Alpha），可以通过Core Graphics创建一个纯RGB格式的上下文，把图像绘制进去后提取像素数据，这样得到的就是width * height * 3大小的UInt8数组，每个元素是0-255的分量值，顺序是R→G→B依次排列。

Swift 实现代码 #

// 先处理图像方向（避免拍照等场景下的旋转问题）
func correctedCGImage(from image: UIImage) -> CGImage? {
    guard let cgImage = image.cgImage else { return nil }
    // 如果图像方向是正常的，直接返回
    if image.imageOrientation == .up {
        return cgImage
    }
    // 否则绘制校正后的图像
    UIGraphicsBeginImageContextWithOptions(image.size, false, image.scale)
    image.draw(in: CGRect(origin: .zero, size: image.size))
    let correctedImage = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    return correctedImage?.cgImage
}

// 提取RGB字节数组
func rgbBytes(from image: UIImage) -> [UInt8]? {
    guard let cgImage = correctedCGImage(from: image) else { return nil }
    
    let width = cgImage.width
    let height = cgImage.height
    let bytesPerPixel = 3 // RGB三通道
    let bytesPerRow = bytesPerPixel * width
    let totalBytes = bytesPerRow * height
    
    // 创建纯RGB格式的上下文（无Alpha通道）
    guard let context = CGContext(data: nil,
                                  width: width,
                                  height: height,
                                  bitsPerComponent: 8, // 每个颜色分量8位（0-255）
                                  bytesPerRow: bytesPerRow,
                                  space: CGColorSpaceCreateDeviceRGB(),
                                  bitmapInfo: CGImageAlphaInfo.none.rawValue) else {
        return nil
    }
    
    // 将图像绘制到上下文
    context.draw(cgImage, in: CGRect(x: 0, y: 0, width: width, height: height))
    
    // 提取并转换为UInt8数组
    guard let pixelData = context.data else { return nil }
    let data = Data(bytes: pixelData, count: totalBytes)
    return [UInt8](data)
}

Objective-C 实现代码 #

// 校正图像方向
- (CGImageRef)correctedCGImageFromImage:(UIImage *)image {
    CGImageRef cgImage = image.CGImage;
    if (!cgImage) return NULL;
    
    if (image.imageOrientation == UIImageOrientationUp) {
        return CGImageRetain(cgImage);
    }
    
    UIGraphicsBeginImageContextWithOptions(image.size, NO, image.scale);
    [image drawInRect:CGRectMake(0, 0, image.size.width, image.size.height)];
    UIImage *correctedImage = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return CGImageRetain(correctedImage.CGImage);
}

// 提取RGB字节数组
- (NSArray<NSNumber *> *)rgbBytesFromImage:(UIImage *)image {
    CGImageRef cgImage = [self correctedCGImageFromImage:image];
    if (!cgImage) return nil;
    
    NSUInteger width = CGImageGetWidth(cgImage);
    NSUInteger height = CGImageGetHeight(cgImage);
    NSUInteger bytesPerPixel = 3;
    NSUInteger bytesPerRow = bytesPerPixel * width;
    NSUInteger totalBytes = bytesPerRow * height;
    
    CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
    CGContextRef context = CGBitmapContextCreate(NULL, width, height, 8, bytesPerRow, colorSpace, kCGImageAlphaNone);
    CGColorSpaceRelease(colorSpace);
    
    if (!context) {
        CGImageRelease(cgImage);
        return nil;
    }
    
    CGContextDrawImage(context, CGRectMake(0, 0, width, height), cgImage);
    UInt8 *pixelData = (UInt8 *)CGBitmapContextGetData(context);
    
    NSMutableArray *rgbArray = [NSMutableArray arrayWithCapacity:totalBytes];
    for (NSUInteger i = 0; i < totalBytes; i++) {
        [rgbArray addObject:@(pixelData[i])];
    }
    
    CGContextRelease(context);
    CGImageRelease(cgImage);
    return rgbArray;
}

注意事项 #

• 图像方向校正很重要：如果是拍照获取的图像，UIImage可能带有旋转的orientation标记，直接提取会导致像素数组的内容和视觉上的图像不一致，所以先做方向校正。
• 输出的数组顺序是R→G→B，每个像素的三个分量连续排列，总长度正好是width * height * 3，完全符合你服务器的要求。

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