OpenCV iOS 图像处理和人脸识别

iOS工程添加OpenCV配置方法如下
https://blog.csdn.net/verybigbug/article/details/113588991

配置好后,就可以在移动端开发OpenCV了。我用的是Swift语言。

1 简单的图片处理

import opencv2可以直接导入OpenCV,不需要写c或者bridging代码。

大部分方法可以用Imgproc直接调,OpenCV的核心图像类Mat可以与iOS的UIImage和CGImage相互转换。

import opencv2

class MyViewController: UIViewController {
    override func viewDidLoad() {
        super.viewDidLoad()
        let image1 = UIImage(named: "001")!
        let iv1 = UIImageView(image: image1)
        let iv2 = UIImageView()
        iv1.frame = CGRect(x: 100, y: 100, width: 200, height: 200)
        iv2.frame = CGRect(x: 100, y: 300, width: 200, height: 200)
        view.addSubview(iv1)
        view.addSubview(iv2)
        
        let m1 = Mat(uiImage: image1)
        Imgproc.cvtColor(src: m1, dst: m1, code: .COLOR_BGRA2GRAY)
        iv2.image = m1.toUIImage()
    }
}

2 使用相机

使用CvVideoCamera2类,设置帧率、尺寸、方向等参数,开启相机,然后在CvVideoCameraDelegate2processImage代理方法中可以获取实时图像。

有几点需要注意:

  • CvVideoCamera2对象要放在类里面,不能放在方法里,否则会被马上回收
  • processImage方法不是主线程,设置图像需要在主线程
  • 图像是BGR格式的要转成RGB,不然你就会发现你的脸是绿的!
import opencv2

class MyViewController: UIViewController, CvVideoCameraDelegate2 {
    
    var lastTime = 0.0
    
    func processImage(_ image: Mat!) {
        Imgproc.cvtColor(src: image, dst: image, code: .COLOR_BGR2RGB)
        
        DispatchQueue.main.async {
            self.camView.image = image.toUIImage()
            print("processImage mat \(image.size()) time \((Date().timeIntervalSince1970 - self.lastTime) * 1000) ms")
            self.lastTime = Date().timeIntervalSince1970
        }
    }
    
    
    let cam = CvVideoCamera2.init()
    lazy var camView = UIImageView(frame: view.frame)
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        camView.contentMode = .scaleAspectFill
        let w = UIScreen.main.bounds.width
        camView.frame = CGRect(x: 0, y: 0, width: w, height: w * 720 / 1280)
        view.addSubview(camView)
        
        cam.delegate = self
        cam.defaultAVCaptureDevicePosition = .front
        cam.defaultAVCaptureSessionPreset = AVCaptureSession.Preset.hd1280x720.rawValue
        cam.defaultAVCaptureVideoOrientation = .portrait
        cam.defaultFPS = 30
        cam.start()
    } 

3 人脸识别

有三种方式,其中两种是OpenCV的:级联分类器和DNN,它们要用模型文件,下载地址我在上一篇中提到了,另一种是iOS自带的CIFilter方式。我分别实现一下。

3.1 级联分类器人脸识别

我用的iOS设备是A12处理器的iPad Mini5,检测时间在70ms左右,每秒只有十几帧,有点卡;用pyrDown将图像缩小后(注释的代码)检测时间提高到33ms左右,明显流畅了,当然检测准确率还是一般。

    let cc_path = Bundle.main.path(forResource: "lbpcascade_frontalface_improved", ofType: "xml")
    lazy var cc = CascadeClassifier.init(filename: cc_path!)


        let gray = Mat()
        Imgproc.cvtColor(src: image, dst: gray, code: .COLOR_BGRA2GRAY)
//        Imgproc.pyrDown(src: gray, dst: gray)
        Imgproc.equalizeHist(src: gray, dst: gray)

        var rects:[Rect2i] = []
        cc.detectMultiScale(image: gray, objects: &rects)
        for r in rects {
//            r.x *= 2
//            r.y *= 2
//            r.width *= 2
//            r.height *= 2
            Imgproc.rectangle(img: image, rec: r, color: Scalar(0, 0, 255, 255), thickness: 2)
        }
      

3.2 DNN 人脸检测

检测效果非常好,检测时间在55ms左右,稍微有点卡,并且缩小图像并不能加快速度。

目前我还没想到能加快计算速度的方法,它应该不支持iOS设备的GPU加速,也许用TensorFlow Lite模型?

    let pb_path = Bundle.main.path(forResource: "opencv_face_detector_uint8", ofType: "pb")
    let pbtxt_path = Bundle.main.path(forResource: "opencv_face_detector", ofType: "pbtxt")
    lazy var net = Dnn.readNetFromTensorflow(model: pb_path!, config: pbtxt_path!)


        let blob = Dnn.blobFromImage(image: image, scalefactor: 1.0, size: Size2i(width: 300, height: 300), mean: Scalar(104,177,123), swapRB: false, crop: false)
        net.setInput(blob: blob)
        let probs = net.forward()

        let probsData = Data.init(bytes: probs.dataPointer(), count: probs.elemSize() * probs.total())
        let detectionMat = Mat(rows: probs.size(2), cols: probs.size(3), type: CvType.CV_32F, data: probsData)

        for i in 0..<detectionMat.rows() {
            let confidence = detectionMat.get(row: i, col: 2)[0]
            if confidence > 0.5 {
                let x1 = Int32(detectionMat.get(row: i, col: 3)[0] * Double(image.cols()))
                let y1 = Int32(detectionMat.get(row: i, col: 4)[0] * Double(image.rows()))
                let x2 = Int32(detectionMat.get(row: i, col: 5)[0] * Double(image.cols()))
                let y2 = Int32(detectionMat.get(row: i, col: 6)[0] * Double(image.rows()))
                let r = Rect2i(x: x1, y: y1, width: x2 - x1, height: y2 - y1)
                Imgproc.rectangle(img: image, rec: r, color: Scalar(0, 0, 255, 255), thickness: 2)
            }
        }

3.3 CIFilter 人脸检测

检测前用CIImage.init(cgImage: image.toCGImage())将Mat转换成CIImage格式

检测时间在33ms左右,比较流畅,检测效果比DNN略差,但是也很准确了。


    lazy var cidetector = CIDetector.init(ofType: CIDetectorTypeFace, context: nil)!


        let features = cidetector.features(in: CIImage.init(cgImage: image.toCGImage()))
        print("processImage ciimage features \(features.count)")
        for f in features {
            let x = Int32(f.bounds.minX)
            let y = Int32(f.bounds.minY)
            let w = Int32(f.bounds.width)
            let h = Int32(f.bounds.height)
            let r = Rect2i(x: x, y: image.height() - y - h, width: w, height: h)
            Imgproc.rectangle(img: image, rec: r, color: Scalar(0, 0, 255, 255), thickness: 2)
        }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,088评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,715评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,361评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,099评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 60,987评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,063评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,486评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,175评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,440评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,518评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,305评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,190评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,550评论 3 298
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,880评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,152评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,451评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,637评论 2 335

推荐阅读更多精彩内容