最近在封装一个手机摄像头测心率的模块，搞得精神各种紧张，导致吃饭、路上、做梦，甚至都在想这个东西，就在刚刚终于搞完了，写个文章庆祝一下。
原理
简单介绍一下，网上可以查到很多关于手机测心率的这种项目，大概就是：把手指放在摄像头和闪光灯上，通过手指处脉搏跳动充血导致的细微颜色变化来确定心跳波动，确定波峰波谷，根据两个波峰之间的时间差来确定瞬时心率。
思路
首先，采集视频流，根据拿到的RGB颜色转成HSV颜色集，其实我们只用到了HSV的H。
对拿到的H进行一些处理，看跟人喜好或者具体情况，主要是用于后面的折线图和计算瞬时心率，如果有能力的话可以处理一下噪音数据，因为可能测的时候手指轻微抖动会造成一些不稳定的数据。
根据处理后的H就可以进行画折线图了，我是把处理后的H和时间戳进行了绑定，用来后面的计算心率。
根据处理后的H来确定波峰波谷，利用两个波谷之间的时间差计算心率。
实现
大致思路就是上面这样，下面来看一下代码具体实现以下。
贴个评论
evolrof 刚刚 你怎么能用HSV呢？H是色相，不管是否有血液流过，色相基本都是红色，可以说是从未改变过。
1.首先我先初始化了一些数据,方便后面使用
//设备@property(strong,nonatomic)AVCaptureDevice*device;//结合输入输出@property(strong,nonatomic)AVCaptureSession*session;//输入设备@property(strong,nonatomic)AVCaptureDeviceInput*input;//输出设备@property(strong,nonatomic)AVCaptureVideoDataOutput*output;//输出的所有点@property(strong,nonatomic)NSMutableArray*points;//记录浮点变化的前一次的值staticfloatlastH=0;//用于判断是否是第一个福点值staticintcount=0;//初始化self.device=[AVCaptureDevicedefaultDeviceWithMediaType:AVMediaTypeVideo];self.session=[[AVCaptureSessionalloc]init];self.input=[[AVCaptureDeviceInputalloc]initWithDevice:self.deviceerror:nil];self.output=[[AVCaptureVideoDataOutputalloc]init];self.points=[[NSMutableArrayalloc]init];
2.设置视频采集流，为了节省内存，我没有输出视频画面
//开启闪光灯if([self.deviceisTorchModeSupported:AVCaptureTorchModeOn]){[self.devicelockForConfiguration:nil];//开启闪光灯self.device.torchMode=AVCaptureTorchModeOn;//调低闪光灯亮度（为了减少内存占用和避免时间长手机发烫）[self.devicesetTorchModeOnWithLevel:0.01error:nil];[self.deviceunlockForConfiguration];}//开始配置inputoutput[self.sessionbeginConfiguration];//设置像素输出格式NSNumber*BGRA32Format=[NSNumbernumberWithInt:kCVPixelFormatType_32BGRA];NSDictionary*setting=@{(id)kCVPixelBufferPixelFormatTypeKey:BGRA32Format};[self.outputsetVideoSettings:setting];//抛弃延迟的帧[self.outputsetAlwaysDiscardsLateVideoFrames:YES];//开启摄像头采集图像输出的子线程dispatch_queue_toutputQueue=dispatch_queue_create("VideoDataOutputQueue",DISPATCH_QUEUE_SERIAL);//设置子线程执行代理方法[self.outputsetSampleBufferDelegate:selfqueue:outputQueue];//向session添加if([self.sessioncanAddInput:self.input])[self.sessionaddInput:self.input];if([self.sessioncanAddOutput:self.output])[self.sessionaddOutput:self.output];//降低分辨率，减少采样率（为了减少内存占用）self.session.sessionPreset=AVCaptureSessionPreset1280x720;//设置最小的视频帧输出间隔self.device.activeVideoMinFrameDuration=CMTimeMake(1,10);//用当前的output初始化connectionAVCaptureConnection*connection=[self.outputconnectionWithMediaType:AVMediaTypeVideo];[connectionsetVideoOrientation:AVCaptureVideoOrientationPortrait];//完成编辑[self.sessioncommitConfiguration];//开始运行[self.sessionstartRunning];
这里我降低了闪光灯亮度，降低了分辨率，减少了每秒钟输出的帧。主要就是为了减少内存的占用。（我手里只有一台6，没有测其他设备可不可以）
3.在output的代理方法中采集视频流
//captureOutput->当前outputsampleBuffer->样本缓冲connection->捕获连接-(void)captureOutput:(AVCaptureOutput*)captureOutputdidOutputSampleBuffer:(CMSampleBufferRef)sampleBufferfromConnection:(AVCaptureConnection*)connection{//获取图层缓冲CVPixelBufferRefimageBuffer=CMSampleBufferGetImageBuffer(sampleBuffer);CVPixelBufferLockBaseAddress(imageBuffer,0);uint8_t*buf=(uint8_t*)CVPixelBufferGetBaseAddress(imageBuffer);size_tbytesPerRow=CVPixelBufferGetBytesPerRow(imageBuffer);size_twidth=CVPixelBufferGetWidth(imageBuffer);size_theight=CVPixelBufferGetHeight(imageBuffer);floatr=0,g=0,b=0;floath,s,v;//计算RGBTORGB(buf,width,height,bytesPerRow,&r,&g,&b);//RGB转HSVRGBtoHSV(r,g,b,&h,&s,&v);//获取当前时间戳（精确到毫秒）doublet=[[NSDatedate]timeIntervalSince1970]*1000;//返回处理后的浮点值floatp=HeartRate(h);//绑定浮点和时间戳NSDictionary*point=@{[NSNumbernumberWithDouble:t]:[NSNumbernumberWithFloat:p]};//下面按个人情况可以进行计算心率或者画心率图}
到这里数据已经处理好了，后面可以根据数据画折线图，或者计算心率
计算RGB
voidTORGB(uint8_t*buf,floatww,floathh,size_tpr,float*r,float*g,float*b){floatwh=(float)(ww*hh);for(inty=0;y<hh;y++){for(intx=0;x<ww*4;x+=4){*b+=buf[x];*g+=buf[x+1];*r+=buf[x+2];}buf+=pr;}*r/=255*wh;*g/=255*wh;*b/=255*wh;}
RGB转HSV
voidRGBtoHSV(floatr,floatg,floatb,float*h,float*s,float*v){floatmin,max,delta;min=MIN(r,MIN(g,b));max=MAX(r,MAX(g,b));*v=max;delta=max-min;if(max!=0)*s=delta/max;else{*s=0;*h=-1;return;}if(r==max)*h=(g-b)/delta;elseif(g==max)*h=2+(b-r)/delta;else*h=4+(r-g)/delta;*h*=60;if(*h<0)*h+=360;}
根据h处理浮点
floatHeartRate(floath){floatlow=0;count++;lastH=(count==1)?h:lastH;low=(h-lastH);lastH=h;returnlow;}
4.分析数据，计算心率
这里我纠结了好长时间，试了几种不同的方法，都没有一个比较理想的结果，计算出来的特别不准。后来看了/88158/这篇文章，后面优化的部分有一个
基音算法，虽不明，但觉厉，对此表示非常感谢。吼吼吼。原理:就是说划一个时间段,在这个时间段里面找到一个
最低峰值,然后确定一个周期，然后分别在这个峰值前间隔0.5个周期的1周期里和这个峰值后间隔0.5个周期的1周期里找到一个最低峰值。 然后根据这几个值来确定瞬时心率。
-(void)analysisPointsWith:(NSDictionary*)point{[self.pointsaddObject:point];if(self.points.countcenterintd_i_l=0;//最低峰值左面的最低峰值位置l->leftintd_i_r=0;//最低峰值右面的最低峰值位置r->rightfloattrough_c=0;//最低峰值的浮点值floattrough_l=0;//最低峰值左面的最低峰值浮点值floattrough_r=0;//最低峰值右面的最低峰值浮点值//1.先确定数据中的最低峰值for(inti=0;i<count;i++){floattrough=[[[self.points[i]allObjects]firstObject]floatValue];if(trough<trough_c){trough_c=trough;d_i_c=i;}}//2.找到最低峰值以后以最低峰值为中心找到前0.5-1.5周期中的最低峰值和后0.5-1.5周期的最低峰值if(d_i_c>=1.5*T){//a.如果最低峰值处在中心位置，即距离前后都至少有1.5个周期if(d_i_cd_i_c-1.5*T;j--){floattrough=[[[self.points[j]allObjects]firstObject]floatValue];if(trough<trough_l){trough_l=trough;d_i_l=j;}}//右面最低峰值for(intk=d_i_c+0.5*T;k<d_i_c+1.5*T;k++){floattrough=[[[self.points[k]allObjects]firstObject]floatValue];if(trough<trough_r){trough_r=trough;d_i_r=k;}}}//b.如果最低峰值右面不够1.5个周期分两种情况不够0.5个周期和够0.5个周期else{//b.1够0.5个周期if(d_i_cd_i_c-1.5*T;j--){floattrough=[[[self.points[j]allObjects]firstObject]floatValue];if(trough<trough_l){trough_l=trough;d_i_l=j;}}//右面最低峰值for(intk=d_i_c+0.5*T;k<count;k++){floattrough=[[[self.points[k]allObjects]firstObject]floatValue];if(trough<trough_r){trough_r=trough;d_i_r=k;}}}//b.2不够0.5个周期else{//左面最低峰值for(intj=d_i_c-0.5*T;j>d_i_c-1.5*T;j--){floattrough=[[[self.points[j]allObjects]firstObject]floatValue];if(trough<trough_l){trough_l=trough;d_i_l=j;}}}}}//c.如果左面不够1.5个周期一样分两种情况够0.5个周期不够0.5个周期else{//c.1够0.5个周期if(d_i_c>0.5*T){//左面最低峰值for(intj=d_i_c-0.5*T;j>0;j--){floattrough=[[[self.points[j]allObjects]firstObject]floatValue];if(trough<trough_l){trough_l=trough;d_i_l=j;}}//右面最低峰值for(intk=d_i_c+0.5*T;k<d_i_c+1.5*T;k++){floattrough=[[[self.points[k]allObjects]firstObject]floatValue];if(trough<trough_r){trough_r=trough;d_i_r=k;}}}//c.2不够0.5个周期else{//右面最低峰值for(intk=d_i_c+0.5*T;k<d_i_c+1.5*T;k++){floattrough=[[[self.points[k]allObjects]firstObject]floatValue];if(trough<trough_r){trough_r=trough;d_i_r=k;}}}}//3.确定哪一个与最低峰值更接近用最接近的一个最低峰值测出瞬时心率60*1000两个峰值的时间差if(trough_l-trough_c<trough_r-trough_c){NSDictionary*point_c=self.points[d_i_c];NSDictionary*point_l=self.points[d_i_l];doublet_c=[[[point_callKeys]firstObject]doubleValue];doublet_l=[[[point_lallKeys]firstObject]doubleValue];NSIntegerfre=(NSInteger)(60*1000)/(t_c-t_l);if(self.frequency)self.frequency(fre);if([self.delegaterespondsToSelector:@selector(startHeartDelegateRateFrequency:)])[self.delegatestartHeartDelegateRateFrequency:fre];}else{NSDictionary*point_c=self.points[d_i_c];NSDictionary*point_r=self.points[d_i_r];doublet_c=[[[point_callKeys]firstObject]doubleValue];doublet_r=[[[point_rallKeys]firstObject]doubleValue];NSIntegerfre=(NSInteger)(60*1000)/(t_r-t_c);if(self.frequency)self.frequency(fre);if([self.delegaterespondsToSelector:@selector(startHeartDelegateRateFrequency:)])[self.delegatestartHeartDelegateRateFrequency:fre];}//4.删除过期数据for(inti=0;i<10;i++){[self.pointsremoveObjectAtIndex:0];}}}
我目前是这样处理的，后面是用的前后两个峰值与最低峰值最接近的那个峰值的时间差，测了几次又和别的app比较了一下，基本都是正确的，最多也就是上下差1-2次每分钟。（在数据比较稳定的情况下，如果有更好的方法请推荐，谢谢）
5.画折线图 这里用到了CoreGraphics
PS:首先，使用这个CoreGraphics要在View里面，并且要在View的drawRect:方法中使用，不然拿不到画布。我是为了封装单独建立了一个UIView的类。
a.首先还是数据，没有数据怎么画
@property(strong,nonatomic)NSMutableArray*points;//在init中初始化数组self.points=[[NSMutableArrayalloc]init];//这个可以翻译过来，也是在init中self.clearsContextBeforeDrawing=YES;//外部调用方法-(void)drawRateWithPoint:(NSNumber*)point{//倒叙插入数组[self.pointsinsertObject:pointatIndex:0];//删除溢出屏幕数据if(self.points.count>self.frame.size.width/6){[self.pointsremoveLastObject];}dispatch_async(dispatch_get_main_queue(),^{//这个方法自动调取drawRect:方法[selfsetNeedsDisplay];});}
之前调setNeedsDisplay,一直没有走drawRect:方法，或者就直走了一次，然后去百度是说setNeedsDisplay会在系统空闲的时候执行drawRect:,然后我尝试着回归到主线程中调用，就好了。具体原因不是很清楚，也可能是因为要在主线程中修改View。
b.画折线的方法，具体怎么调整看个人心情了。
CGFloatww=self.frame.size.width;CGFloathh=self.frame.size.height;CGFloatpos_x=ww;CGFloatpos_y=hh/2;//获取当前画布CGContextRefcontext=UIGraphicsGetCurrentContext();//折线宽度CGContextSetLineWidth(context,1.0);//消除锯齿//CGContextSetAllowsAntialiasing(context,false);//折线颜色CGContextSetStrokeColorWithColor(context,[UIColorredColor].CGColor);CGContextMoveToPoint(context,pos_x,pos_y);for(inti=0;i<self.points.count;i++){floath=[self.points[i]floatValue];pos_y=hh/2+(h*hh/2);CGContextAddLineToPoint(context,pos_x,pos_y);pos_x-=6;}CGContextStrokePath(context);
c.为了看起来好看，我还加了网格，当然也是在drawRect:中调用的
staticCGFloatgrid_w=30.0f;-(void)buildGrid{CGFloatwight=self.frame.size.width;CGFloatheight=self.frame.size.height;//获取当前画布CGContextRefcontext=UIGraphicsGetCurrentContext();CGFloatpos_x=0.0f;CGFloatpos_y=0.0f;//在wight范围内画竖线while(pos_x<wight){//设置网格线宽度CGContextSetLineWidth(context,0.2);//设置网格线颜色CGContextSetStrokeColorWithColor(context,[UIColorgreenColor].CGColor);//起点CGContextMoveToPoint(context,pos_x,1.0f);//终点CGContextAddLineToPoint(context,pos_x,height);pos_x+=grid_w;//开始划线CGContextStrokePath(context);}//在height范围内画横线while(pos_y<height){CGContextSetLineWidth(context,0.2);CGContextSetStrokeColorWithColor(context,[UIColorgreenColor].CGColor);CGContextMoveToPoint(context,1.0f,pos_y);CGContextAddLineToPoint(context,wight,pos_y);pos_y+=grid_w;CGContextStrokePath(context);}pos_x=0.0f;pos_y=0.0f;//在wight范围内画竖线while(pos_x<wight){CGContextSetLineWidth(context,0.1);CGContextSetStrokeColorWithColor(context,[UIColorgreenColor].CGColor);CGContextMoveToPoint(context,pos_x,1.0f);CGContextAddLineToPoint(context,pos_x,height);pos_x+=grid_w/5;CGContextStrokePath(context);}//在height范围内画横线while(pos_y<height){CGContextSetLineWidth(context,0.1);CGContextSetStrokeColorWithColor(context,[UIColorgreenColor].CGColor);CGContextMoveToPoint(context,1.0f,pos_y);CGContextAddLineToPoint(context,wight,pos_y);pos_y+=grid_w/5;CGContextStrokePath(context);}}
总结
没有图，没有效果，只有思路和代码，我把封装好的代码放在了GitHub上，地址：/YvanLiu/HeartBeatsPlugin.git
写这个功能的时候，自己有很多思考，也参考了很多其他人的博客、代码还有别人的毕业论文，呵呵呵，还问了几个学医的同学，代码不难，数据处理的部分可能不太好弄，但是写完还是有点成就感的。
代码里还存在很多问题，心率的计算还存在很高的错误率，现在正在想办法改进，欢迎指正，欢迎批评，欢迎提问。
文章转自YvanLiu的简书
如果觉得《iOS手机摄像头测心率》对你有帮助，请点赞、收藏，并留下你的观点哦！
