iOS下OpenCV开发用OC还是Swift

本文为作者原创,转载请注明出处(http://www.cnblogs.com/mar-q/)by 负赑屃

 

  其实标题中这个问题并不准确,准确的说法应该是iOS下的OpenCV开发是使用OC还是Swift+OC。这个问题纠结了很久,研究了很多例子。先说结论:如果用到的算法规模不大且不熟悉cap_ios.h尽量用Swift+OC。(欢迎高手来打脸)

  iOS下OpenCV开发的例子很多,大家可以直接去GitHub上扒拉,但是Swift还是比较少,先贡献几个能运行的例子:Objective-C(《Instant OpenCV for iOS》的例程,作者是 Alexander Shishkov和 Kirill Kornyakov,OpenCV的活跃分子),Swift(作者是Hiroki Ishiura,一个日本的程序员老哥,饱受秃顶的困扰。。。)16年Joseph Howse出版了一本书《iOS Application Development with OpenCV 3》,算是小白的入门教程吧(目前无中文,正版很贵,而且比较难下到电子书,Google图书里有部分英文电子版),第一章就提高了关于language的选择,由于opencv的核心是用C++写的,Swift不能直接调用C++,需要通过Objective-C作为中间层,所以作者在书中的例程都是基于Objective-C。

一、Objective-C下开发OpenCV的基本流程

  熟悉Android下OpenCV开发的都知道,OpenCV提供了封装好的JavaCameraView和NativeCameraView两个类,连接摄像头时需要通过类实例进行初始化设置(如设置画面帧大小、帧率等信息),在iOS的开发中,OpenCV也提供了cap_ios.h,对摄像头的初始化通过CvVideoCamera类来实现。和Android开发类似,CvVideoCamera也对摄像头的初始化进行了封装。

@interface ViewController : UIViewController<CvVideoCameraDelegate>{
    CvVideoCamera *videoCamera;
}
@property (nonatomic, retain) CvVideoCamera* videoCamera;
@property (weak, nonatomic) IBOutlet UIImageView *imgView;

 

  如代码所示:OC中需要在.h头文件中定义CvVideoCamera,ViewController继承UIViewController,可以看出CvVideoCameraDelegate是一个协议。

@class CvVideoCamera;
@protocol CvVideoCameraDelegate <NSObject>
#ifdef __cplusplus
// delegate method for processing image frames
- (void)processImage:(cv::Mat&)image;
#endif
@end

  通过cap_ios.h可以看出,处理画面帧时只需要重写processImage方法。

- (void)viewDidLoad {
    [super viewDidLoad];
    self.videoCamera = [[CvVideoCamera alloc] initWithParentView:imgView];self.videoCamera.delegate = self;
    self.videoCamera.defaultAVCaptureDevicePosition =
    AVCaptureDevicePositionFront;
    self.videoCamera.defaultAVCaptureSessionPreset =
    AVCaptureSessionPreset640x480;
    self.videoCamera.defaultAVCaptureVideoOrientation =
    AVCaptureVideoOrientationPortrait;
}
- (void)processImage:(cv::Mat&)image{
     cv::cvtColor(image, image, CV_BGR2GRAY);  
}

  在ViewController.m的viewDidLoad方法中对摄像头进行初始化。如需要对画面帧进行处理重写processImage方法即可,有点类似Android中onCameraViewStarted的回调方法(上面代码通过OpenCV中的cv::cvtColor方法,实现了摄像头的灰度模式)。 

二、Swift下开发OpenCV的基本流程

  由于Swift不能直接调用C++,所有对C++的调用需要经过OC来包裹处理,所以Swift使用OpenCV需要和OC进行混编,关于Swift+OC+OpenCV的混编流程大概描述一下:

  1、新建Swift项目;

  2、在Swift项目中新建ObjectiveC文件,后缀为.m的请修改为.mm,此时会提示是否生成bridgeXXX.h的头文件,确认生成;

  3、为.mm文件新建一个头文件,并在生成的bridgeXXX.h中import该头文件;

  4、至此在.mm文件中已经可以调用OpenCV中的方法,如果你需要cpp或者hpp文件,请记住,它们不能和Swift直接交互,必须通过OC作为桥梁。

  由于在Swift代码中无法直接使用CvVideoCamera,初始化摄像头需要通过AVFoundation来实现。

import AVFoundation
class ViewController: UIViewController, AVCaptureVideoDataOutputSampleBufferDelegate {//粗体为AVFoundation中的delegate
    @IBOutlet weak var imageView: UIImageView!
    var session: AVCaptureSession!
    var device: AVCaptureDevice!
    var output: AVCaptureVideoDataOutput!
    override func viewDidLoad() {
        super.viewDidLoad()
        // Prepare a video capturing session.
        self.session = AVCaptureSession()
        self.session.sessionPreset = AVCaptureSessionPreset640x480  
    ……  
  }
……

   如上代码所示,在Swift初始化摄像头的过程中没有用到OpenCV的任何东西。如果需要对画面帧进行处理,需要实现AVCaptureVideoDataOutputSampleBufferDelegate的captureOutput方法,从命名上可以看出这也是一个Delegate(协议),其中有一个captureOutput方法(代码如下),可以在方法内处理捕捉到的实时画面帧。

    func captureOutput(_ captureOutput: AVCaptureOutput!, didOutputSampleBuffer sampleBuffer: CMSampleBuffer!, from connection: AVCaptureConnection!) {
        // 将捕捉到的image buffer 转换成 UIImage.
        guard let buffer: CVPixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else {
            print("could not get a pixel buffer")
            return
        }
        let capturedImage: UIImage
        do {
            CVPixelBufferLockBaseAddress(buffer, CVPixelBufferLockFlags.readOnly)
            defer {
                CVPixelBufferUnlockBaseAddress(buffer, CVPixelBufferLockFlags.readOnly)
            }
            let address = CVPixelBufferGetBaseAddressOfPlane(buffer, 0)
            let bytes = CVPixelBufferGetBytesPerRow(buffer)
            let width = CVPixelBufferGetWidth(buffer)
            let height = CVPixelBufferGetHeight(buffer)
            let color = CGColorSpaceCreateDeviceRGB()
            let bits = 8
            let info = CGBitmapInfo.byteOrder32Little.rawValue | CGImageAlphaInfo.premultipliedFirst.rawValue
            guard let context = CGContext(data: address, width: width, height: height, bitsPerComponent: bits, bytesPerRow: bytes, space: color, bitmapInfo: info) else {
                print("could not create an CGContext")
                return
            }
            guard let image = context.makeImage() else {
                print("could not create an CGImage")
                return
            }
            capturedImage = UIImage(cgImage: image, scale: 1.0, orientation: UIImageOrientation.up)
       //调用你写在.mm文件中的OpenCV方法 } }

 

  代码中的captureOutput方法捕捉到了实时的画面帧,通过提取buffer中的画面帧并将其转换为UIImage,为了实现和OC中同样的灰度画面,需要另外定义一个ViewProcress.mm和ViewProcress.h,通过ViewProcress.mm中的OpenCV方法来处理你捕捉到的UIImage,将其转换为灰度模式。

三、对比

  从代码来看,Swift是在曲线救国,为了验证,我写了一个简单的高斯背景建模方法处理实时画面帧cv::BackgroundSubtractorMOG2,这个方法的实时开销在OpenCV的常用方法中应该仅次于光流法,在两段代码中我都将帧率设置为30FPS。通过实际运行来看,Swift平均帧率达到30FPS,而OC只有15FPS。主要原因有两点:

  1. 并发编程,OC中没用并发,而在Swift中使用了DispatchQueue,这得益于Swift3带来的改变,处理画面帧的时候需要并发,但是不能随便并发,而是要在一个队列中,所以就用到了DispatchQueue,这是Swift下帧率高的主要原因,你可能会说,那我在OC下也用多线程就好了,OK,请看下一个点分析。

  2. OC中的摄像头是通过OpenCV封装的协议来初始化的,高斯背景建模方法直接作用于画面帧(processImage方法已经将实时画面帧转换为Mat,可以直接处理)。而Swift的是直接用AVFoundation初始化摄像头,captureOutput方法获取的是UIImage,需要在.mm文件中先将UIImage转换为Mat再调用高斯背景建模方法,处理完成再返回给UIView。关于帧率的初始化,OpenCV只提供了一个defaultFPS的设置方法,如果帧率超出范围,反而会衰减的更厉害,而在Swift中因为无法直接使用OpenCV提供的方法,所以初始化帧率必须通过AVFoundation的CMTimeMake函数来处理,设置一个最小帧率,当帧率无法满足时系统会自动平衡。这是帧率稳定的主要原因。

  特别提示:注意内存管理!注意内存管理!注意内存管理!重要的事情说三遍。如果你用Swift来写,那么就会涉及Swift、Objective-C、C++。。。内存怎么办。。。有几点建议:一、使用完的mat注意release掉;二、多用vector少用数组;三、非计算尽量面向对象,专用计算尽量静态方法;四、如果可以,除了需要调用的库外,不要用C++,尽量用Objective-C。如果你有更好的办法,请指点。。。

  现在回到开头的结论,其实在OC中也可以使用AVFoundation来初始化摄像头,但是为了偷懒,相信大多数人都会直接用OpenCV自带的方法完成初始化。而OpenCV封装类对Android或iOS的系统是以兼容为主的,效率是其次的,所以如果在实际开发中,不论Android还是iOS,都建议使用系统推荐的方式来调用camera,只有在处理画面帧时才调用OpenCV中的方法。而且随着Swift越来越完善,当然是推荐使用Swift+OC来进行OpenCV下的开发了。 

posted @ 2017-04-05 23:17  负赑屃  阅读(5007)  评论(3编辑  收藏  举报