Obj-C实现图像颜色替换（绿幕）出错：画面整体褪色而非仅替换目标色

我用Objective-C开发的应用要实现绿幕功能：选择图片后把指定天蓝色替换成星空背景图，但现在处理后画面整体褪色，几乎所有内容都被替换了，不是只替换目标天空颜色。我知道有Swift库能实现，但必须用Obj-C。

原实现代码 #

ImageProcessor类代码 #

+ (UIImage *)replaceWhiteColorWithImage:(UIImage *)image backgroundImage:(UIImage *)backgroundImage {
    CGImageRef rawImageRef = image.CGImage;
    const CGFloat colorMasking[6] = {130/255.0, 1.0, 151/255.0, 1.0, 186/255.0, 1.0};
    UIGraphicsBeginImageContext(image.size);

    CGContextRef context = UIGraphicsGetCurrentContext();

    // Flip the context to match the UIKit coordinate system
    CGContextTranslateCTM(context, 0.0, image.size.height);
    CGContextScaleCTM(context, 1.0, -1.0);

    // Draw the background image
    [backgroundImage drawInRect:CGRectMake(0, 0, image.size.width, image.size.height)];

    // Clip to the raw image using the white color as a mask
    CGContextClipToMask(context, CGRectMake(0, 0, image.size.width, image.size.height), rawImageRef);

    // Clear the white color with a transparent background
    CGContextClearRect(context, CGRectMake(0, 0, image.size.width, image.size.height));

    UIImage *result = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    return result;
}

ViewController调用代码 #

-(IBAction)chooseImage {
    UIImagePickerController *imagePicker = [[UIImagePickerController alloc] init];
    imagePicker.delegate = self;
    [self presentViewController:imagePicker animated:YES completion:nil];
}

- (void)imagePickerController:(UIImagePickerController *)picker didFinishPickingMediaWithInfo:(NSDictionary<UIImagePickerControllerInfoKey,id> *)info {
    
    UIImage *pickedImage = info[UIImagePickerControllerOriginalImage];
    UIImage *backgroundImage = [UIImage imageNamed:@"space.jpg"];
    UIImage *processedImage = [ImageProcessor replaceWhiteColorWithImage:pickedImage backgroundImage:backgroundImage];
    
    self.yourImageView.image = processedImage;
    [picker dismissViewControllerAnimated:YES completion:nil];
}

相关图片 #

• 待编辑原图：
• 目标背景图：
• 错误结果图：

错误原因分析 #

1. 颜色掩码未实际使用：你定义了colorMasking数组，但全程没调用CGImageCreateWithMaskingColors来生成颜色遮罩，等于这个掩码完全没起作用。
2. 遮罩逻辑完全错误：当前代码用原始图片作为遮罩，然后清空整个遮罩区域，这会把原图所有内容都删除，只留下背景图，和“替换指定颜色”的需求完全相反。
3. 颜色范围过于宽泛：就算用上掩码，你设置的RGB范围（130/255到1，151/255到1，186/255到1）覆盖了大量非天蓝色区域，会导致误替换。

修正后的实现代码 #

改进后的ImageProcessor方法 #

+ (UIImage *)replaceSkyColorWithBackgroundImage:(UIImage *)sourceImage backgroundImage:(UIImage *)backgroundImage {
    // 创建适配Retina的绘图上下文，保证图片清晰度
    UIGraphicsBeginImageContextWithOptions(sourceImage.size, NO, sourceImage.scale);
    CGContextRef context = UIGraphicsGetCurrentContext();
    
    // 翻转上下文以匹配UIKit的坐标系
    CGContextTranslateCTM(context, 0, sourceImage.size.height);
    CGContextScaleCTM(context, 1.0, -1.0);
    
    // 先绘制背景图
    [backgroundImage drawInRect:CGRectMake(0, 0, sourceImage.size.width, sourceImage.size.height)];
    
    // 精准定义天蓝色的颜色范围（根据原图调整，避免误替换）
    // 原图天蓝色的RGB大致范围：R(130-160), G(151-180), B(186-210)，转换为0-1的浮点数
    const CGFloat colorMasking[6] = {130/255.0, 160/255.0, 151/255.0, 180/255.0, 186/255.0, 210/255.0};
    
    // 生成颜色遮罩：目标天蓝色区域变为透明，其他颜色保留
    CGImageRef maskedImageRef = CGImageCreateWithMaskingColors(sourceImage.CGImage, colorMasking);
    UIImage *maskedImage = [UIImage imageWithCGImage:maskedImageRef scale:sourceImage.scale orientation:sourceImage.imageOrientation];
    CGImageRelease(maskedImageRef); // 释放CGImage资源，避免内存泄漏
    
    // 绘制处理后的原图（透明区域会透出下方的背景图）
    [maskedImage drawInRect:CGRectMake(0, 0, sourceImage.size.width, sourceImage.size.height)];
    
    // 获取最终结果并清理上下文
    UIImage *result = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return result;
}

修正后的ViewController调用代码 #

- (void)imagePickerController:(UIImagePickerController *)picker didFinishPickingMediaWithInfo:(NSDictionary<UIImagePickerControllerInfoKey,id> *)info {
    UIImage *pickedImage = info[UIImagePickerControllerOriginalImage];
    UIImage *backgroundImage = [UIImage imageNamed:@"space.jpg"];
    
    // 调用改进后的方法
    UIImage *processedImage = [ImageProcessor replaceSkyColorWithBackgroundImage:pickedImage backgroundImage:backgroundImage];
    
    self.yourImageView.image = processedImage;
    [picker dismissViewControllerAnimated:YES completion:nil];
}

关键改进点 #

• 正确使用颜色掩码：通过CGImageCreateWithMaskingColors生成仅保留非目标颜色的图片，目标区域变透明，实现背景替换效果。
• 精准颜色范围：缩小RGB范围，只匹配原图中的天蓝色区域，减少误替换。
• 适配Retina屏：使用UIGraphicsBeginImageContextWithOptions替代旧方法，保证处理后的图片不失真。
• 逻辑修正：先画背景图，再画带透明区域的原图，符合“替换指定颜色为背景”的需求。

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