iOS AVFoundation-人脸识别

关于AVFoundation介绍"http://www.jianshu.com/p/8c5c150dba65"

Unknown.jpeg

导入头文件

#import <AVFoundation/AVFoundation.h>

遵循代理协议

<AVCaptureMetadataOutputObjectsDelegate>
//捕捉会话
@property(nonatomic,strong)AVCaptureSession *captureSession;
//展示layer
@property (nonatomic,strong)AVCaptureVideoPreviewLayer *videoPreviewLayer;
//保存屏幕中离开的人脸
@property(nonatomic,strong)NSMutableArray *leaveFaceArray;
//保存屏幕检测到脸对应的layer faceID 作为字典 key
@property(nonatomic,strong)NSMutableDictionary *faceLayerDic;

-(NSMutableDictionary *)faceLayerDic{
    if (!_faceLayerDic) {
        
        _faceLayerDic = [NSMutableDictionary dictionaryWithCapacity:10];
    }
    
    return _faceLayerDic;
}

-(void)viewWillAppear:(BOOL)animated{
    
    [super viewWillAppear:animated];
    
     [self startReading];
    
}

检测摄像头数量

- (NSUInteger)cameraCounts {
    
    return [[AVCaptureDevice devicesWithMediaType:AVMediaTypeVideo] count];
}

获取前置摄像头

- (AVCaptureDevice *)inactiveCamera {
    AVCaptureDevice *deviceSelect = nil;
    if (self.cameraCounts > 1) {
        
        NSArray *devices = [AVCaptureDevice devicesWithMediaType:AVMediaTypeVideo];
        for (AVCaptureDevice *device in devices) {
            if (device.position == AVCaptureDevicePositionFront) {
               
                deviceSelect = device;
            }
        }
    }
    
    return deviceSelect;
}

开始识别

-(void)startReading{
    
    //读取摄像头授权状态
    NSString *mediaType = AVMediaTypeVideo;//读取媒体类型
    AVAuthorizationStatus authStatus = [AVCaptureDevice authorizationStatusForMediaType:mediaType];//读取设备授权状态
    if(authStatus == AVAuthorizationStatusRestricted || authStatus == AVAuthorizationStatusDenied){
     
        // 提示开启权限
        
        return;
    }
  
    NSError *error;
    
    //1.初始化捕捉设备(AVCaptureDevice),类型为AVMediaTypeVideo
    AVCaptureDevice *captureDevice = [self inactiveCamera];

    
    //2.用captureDevice创建输入流,输入设备转换成AVCaptureDeviceInput对象
    AVCaptureDeviceInput *input = [AVCaptureDeviceInput deviceInputWithDevice:captureDevice error:&error];
    
    if (!input) {
        NSLog(@"%@", [error localizedDescription]);
      
        return ;
    }
    //3.创建媒体数据输出流
    AVCaptureMetadataOutput *captureMetadataOutput = [[AVCaptureMetadataOutput alloc] init];
    
    //4.实例化捕捉会话
    self.captureSession = [[AVCaptureSession alloc] init];
    
    self.captureSession.sessionPreset = AVCaptureSessionPresetHigh;
    
    //4.1.将输入设备添加到会话
    if ([self.captureSession canAddInput:input]) {

        [self.captureSession addInput:input];
    }
    //4.2.将媒体输出设备添加到会话
    if ([self.captureSession canAddOutput:captureMetadataOutput]) {
        
        [self.captureSession addOutput:captureMetadataOutput];
        
    }
    
    NSArray *metaDataObjectTypes = @[AVMetadataObjectTypeFace];
    //4.3.摄像头在捕捉数据时,只会对人脸元数据感兴趣
    captureMetadataOutput.metadataObjectTypes = metaDataObjectTypes;
   
    //获得主队列,因为人脸检测用到硬件加速,而且很多任务都在主线程执行
    dispatch_queue_t mainQueue = dispatch_get_main_queue();
    
    //设置代理 主队列
    [captureMetadataOutput setMetadataObjectsDelegate:self queue:mainQueue];
    
    
    //5.实例化预览图层
    _videoPreviewLayer = [[AVCaptureVideoPreviewLayer alloc] initWithSession:self.captureSession];
    
    //6.设置预览图层填充方式
    [_videoPreviewLayer setVideoGravity:AVLayerVideoGravityResizeAspectFill];
    
    //7.设置图层的frame
    [_videoPreviewLayer setFrame:self.view.bounds];
    
    [UIView animateWithDuration:2 animations:^{
        
        [self.view.layer addSublayer:_videoPreviewLayer];
        
    }];
    
    //开启会话
    [self.captureSession startRunning];
 
}

代理方法 获取捕获的数据

- (void)captureOutput:(AVCaptureOutput *)captureOutput
didOutputMetadataObjects:(NSArray *)metadataObjects
       fromConnection:(AVCaptureConnection *)connection {
    
  NSMutableArray *faceArray = [NSMutableArray arrayWithCapacity:10];
    
    //获取的人脸放到数组中
    
    for (AVMetadataFaceObject *face in metadataObjects) {
        
    //NSLog(@"faceID:%li",(long)face.faceID);
   // NSLog(@"face.bounds:%@",NSStringFromCGRect(face.bounds));
        
        //将摄像头捕捉的人脸位置转换到屏幕位置
        AVMetadataObject *tranformFace = [_videoPreviewLayer  transformedMetadataObjectForMetadataObject:face];
        
        [faceArray addObject:tranformFace];
    }

    //将获取的人脸数据进行处理
    [self faceOperation:faceArray];

}


-(void)faceOperation:(NSArray *)faceArray{
    
   NSMutableArray *leaveFaceArray = [self.faceLayerDic.allKeys mutableCopy];
    
    for (AVMetadataFaceObject *face in faceArray) {
     
        NSNumber *faceID = @(face.faceID);
        [leaveFaceArray removeObject:faceID];
        CALayer *layer = self.faceLayerDic[faceID];
     
        if(!layer){
           
          //makeFacelayer :新建一个人脸图层
            layer = [self makeFaceLayer];
        
            //将人脸图层添加到videoPreviewLayer
         
            [self.videoPreviewLayer addSublayer:layer];
            
            //将 layer加入字典中
            self.faceLayerDic[faceID] = layer;
        }
        //指定图层的位置
        layer.frame = face.bounds;
    
    }
  
    for (NSNumber *faceID in leaveFaceArray) { 
        CALayer *layer = self.faceLayerDic[faceID];
        [layer removeFromSuperlayer];
        [self.faceLayerDic removeObjectForKey:faceID];
        
    }
}


- (CALayer *)makeFaceLayer {
    CALayer *layer = [CALayer layer];
    layer.borderWidth = 2.0f;
    layer.borderColor = [UIColor redColor].CGColor;
    return layer;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,968评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,601评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,220评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,416评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,425评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,144评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,432评论 3 401
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,088评论 0 261
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,586评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,028评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,137评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,783评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,343评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,333评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,559评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,595评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,901评论 2 345

推荐阅读更多精彩内容