带 alpha 通道的 UIImage 和 cv::Mat 转换

Question

我正在使用以下代码将UIImage*和cv::Mat相互转换：

- (cv::Mat)cvMatFromUIImage:(UIImage *)image
{
  CGColorSpaceRef colorSpace = CGImageGetColorSpace(image.CGImage);
  CGFloat cols = image.size.width;
  CGFloat rows = image.size.height;

  cv::Mat cvMat(rows, cols, CV_8UC4); // 8 bits per component, 4 channels (color channels + alpha)

  CGContextRef contextRef = CGBitmapContextCreate(cvMat.data,                 // Pointer to  data
                                                 cols,                       // Width of bitmap
                                                 rows,                       // Height of bitmap
                                                 8,                          // Bits per component
                                                 cvMat.step[0],              // Bytes per row
                                                 colorSpace,                 // Colorspace
                                                 kCGImageAlphaNoneSkipLast |
                                                 kCGBitmapByteOrderDefault); // Bitmap info flags

  CGContextDrawImage(contextRef, CGRectMake(0, 0, cols, rows), image.CGImage);
  CGContextRelease(contextRef);

  return cvMat;
}

并且

-(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;
 }

我从OpenCV文档中获取了这些内容。我使用它们的方式如下：

UIImage *img = [UIImage imageNamed:@"transparent.png"];
UIImage *img2 = [self UIImageFromCVMat:[self cvMatFromUIImage:img]];

然而，这些功能会丢失alpha通道信息。我知道这是由于标志kCGImageAlphaNone和kCGImageAlphaNoneSkipLast导致的，不幸的是，我无法找到一种方法通过更改这些标志来不丢失alpha信息。

那么，如何在这两种类型之间转换而不丢失alpha信息？

这是我使用的图像：

- guneykayim

---

非常好的问题，@guneykayim。你找到答案了吗？ - atereshkov

Answer 1

我们应该使用来自opencv v2.4.6的这些函数：

UIImage* MatToUIImage(const cv::Mat& image);
void UIImageToMat(const UIImage* image, cv::Mat& m, bool alphaExist = false);

不要忘记包括以下内容：

请注意：

opencv2/imgcodecs/ios.h

- atereshkov

---

你好。你能帮忙吗？我正在使用UIImageToMat，alphaExist = true，但是在透明度方面无法正常工作。 - famfamfam

Answer 2

你需要传递 (kCGBitmapByteOrder32Host | kCGImageAlphaPremultipliedFirst) 而不是 kCGImageAlphaNoneSkipLast，以获取 BGRA 格式的预乘 alpha。CoreGraphics 仅支持预乘 alpha。但是，你需要检查 OpenCV 如何表示像素中的 alpha，以确定如何告诉 OpenCV 像素已经预乘。我使用的代码假设在 OpenCV 中使用了直接 alpha，所以你需要小心处理这个问题。

- MoDJ