Конвертировать UIImage в cv::Mat в OpenCV
Я пытаюсь преобразовать UIImage в cv::Mat, чтобы я мог использовать класс LineIterator OpenCV на нем. Я использую код, предоставленный здесь документацией opencv, в частности, cvMatGrayFromUIImage, найденный на этой странице. Я использовал этот код в функции, которую я написал, а затем вызывал его в моих файлах swift. Однако, когда я пытаюсь напечатать изображение cv::Mat, числа в массиве не точно отражают градации серого входного изображения. Я помещаю массив в Matlab и вызываю функцию imagesc, чтобы увидеть, действительно ли это шкала серого.
Вот код, который я использую в своем OpenCVWrapper.mm
-(void) getPixelIntensity: (UIImage *)image
{
//Transform UIImage to cv::Mat
CGColorSpaceRef colorSpace = CGImageGetColorSpace(image.CGImage);
CGFloat cols = image.size.width;
CGFloat rows = image.size.height;
cv::Mat cvMat(rows, cols, CV_8UC1); // 8 bits per component, 1 channels
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);
//Print image matrix
cout << cvMat;
}
Тогда в моем Viewcontroller.swift
Я называю функцию так
import UIKit
class ViewController: UIViewController{
//MARK: Properties
@IBOutlet weak var photoImageView: UIImageView!
override func viewDidLoad() {
super.viewDidLoad()
override func didReceiveMemoryWarning() {
super.didReceiveMemoryWarning()
// Dispose of any resources that can be recreated.
}
@IBAction func getCoordinates(_ sender: UIButton) {
let opencvwrapper = OpenCVWrapper()
opencvwrapper.getPixelIntensity(photoImageView.image)
}
}
1 ответ
OpenCV предоставляет некоторые функции, которые позволяют вам конвертировать UIImage в cv::Mat и наоборот.
1. Импортируйте библиотеки из заголовка вашей оболочки:
#import <opencv2/imgcodecs/ios.h>
2. Теперь у вас есть доступ к этим функциям:
UIImageToMat(uiImage, imageMat);
UIImage* img = MatToUIImage(imageMat);