从Mat(OpenCV)生成透明度的PNG

时间:2015-04-08 20:12:03

标签: c++ ios opencv image-processing png

我正在做一些图像处理,在透明背景上创建图像。我正在使用OpenCV(C ++)和iOS。首先,我设置了一个4通道8位矩阵,用alpha 0将所有像素设置为白色(这是正确的吗?)。然后将像素从result复制到foreground

 cv::Mat foreground(image.size(),CV_8UC4, cv::Scalar(255,255,255,0));
 image.copyTo(foreground, result);

接下来,我需要将此Mat转换为UIImage,我正在使用的代码:

+(UIImage *)UIImageFromCVMat:(cv::Mat)cvMat
{
    NSData *data = [NSData dataWithBytes:cvMat.data length:cvMat.elemSize()*cvMat.total()];
    CGColorSpaceRef colorSpace;

    if (cvMat.elemSize() == 1) {
        colorSpace = CGColorSpaceCreateDeviceGray();
    } else {
        colorSpace = CGColorSpaceCreateDeviceRGB();
    }

    CGDataProviderRef provider = CGDataProviderCreateWithCFData((__bridge CFDataRef)data);

    // Creating CGImage from cv::Mat
    CGImageRef imageRef = CGImageCreate(cvMat.cols,                                 //width
                                        cvMat.rows,                                 //height
                                        8,                                          //bits per component
                                        8 * cvMat.elemSize(),                       //bits per pixel
                                        cvMat.step[0],                            //bytesPerRow
                                        colorSpace,                                 //colorspace
                                        kCGImageAlphaNone|kCGBitmapByteOrderDefault,// bitmap info
                                        provider,                                   //CGDataProviderRef
                                        NULL,                                       //decode
                                        false,                                      //should interpolate
                                        kCGRenderingIntentDefault                   //intent
                                        );

    // Getting UIImage from CGImage
    UIImage *finalImage = [UIImage imageWithCGImage:imageRef];
    CGImageRelease(imageRef);
    CGDataProviderRelease(provider);
    CGColorSpaceRelease(colorSpace);

    return finalImage;
}

这适用于生成我的图像,但我注意到它不透明。所以我想改为使用CGImageCreate,我可以使用:

CGImageRef imageRef = CGImageCreateWithPNGDataProvider(provider, NULL, false, kCGRenderingIntentDefault);

但是我不确定CGImageCreateWithPNGDataProvider是否是正确的调用,我不确定该方法是否正确格式化了提供程序。

我还检查过以确认我的imageView.opaque = NO;

有没有人知道将cv::Mat转换为PNG的最简单方法?感谢

*编辑:我还试图修改原始CGImageCreate函数上的位图信息的一些标志:

kCGImageAlphaPremultipliedLast,  /* For example, premultiplied RGBA */
  kCGImageAlphaPremultipliedFirst, /* For example, premultiplied ARGB */
  kCGImageAlphaLast,               /* For example, non-premultiplied RGBA */
  kCGImageAlphaFirst,              /* For example, non-premultiplied ARGB */
  kCGImageAlphaNoneSkipLast,       /* For example, RBGX. */
  kCGImageAlphaNoneSkipFirst,      /* For example, XRGB. */

但是,由于我使用4个频道初始化了<Error>: CGImageCreate: invalid image bits/pixel: 24.,因此我收到错误cv::Mat foreground我不确定如何解释。

1 个答案:

答案 0 :(得分:1)

看起来问题是 copyTo 方法。来自doc:

  

该方法将矩阵数据复制到另一个矩阵。在复制数据之前,该方法调用

     

m.create(this-&gt; size(),this-&gt; type());

它似乎会覆盖您的前景的设置。你需要找到另一种方法来复制比特以获得alpha通道。