Конвертировать 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);
Другие вопросы по тегам