我正在使用以下代码将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 Documentation获取了这些内容。我按如下方式使用它们:
UIImage *img = [UIImage imageNamed:@"transparent.png"];
UIImage *img2 = [self UIImageFromCVMat:[self cvMatFromUIImage:img]];
然而,这些功能会丢失Alpha通道信息。我知道这是因为标志kCGImageAlphaNone
和kCGImageAlphaNoneSkipLast
,遗憾的是我找不到通过更改这些标志而丢失alpha信息的方法。
那么,如何在不丢失alpha信息的情况下在彼此之间转换这两种类型?
以下是我使用的图片:
答案 0 :(得分:1)
您不需要传递kCGImageAlphaNoneSkipLast
而是传递(kCGBitmapByteOrder32Host | kCGImageAlphaPremultipliedFirst)
以获得BGRA格式的预乘alpha。 CoreGraphics仅支持预乘alpha。但是,您需要检查OpenCV如何以像素为单位表示alpha,以确定如何告诉OpenCV像素已经预乘。我使用的代码假设使用OpenCV的直接alpha,所以你需要小心。
答案 1 :(得分:0)
我们应该使用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