iOS 矩形识别

需求:识别图片中的矩形框,标出矩形区域,并可以裁剪矩形区域
实现方式:
1、使用CIDetector类识别矩形区域

ps:也可以使用Vision(仅支持iOS11.0以上版本)框架的的VNDetectRectanglesRequest进行矩形框识别

当前使用的detectorType为CIDetectorTypeRectangle(矩形识别),可以通过修改其他type进行面部识别、二维码识别、文本识别

// Types to be used for +[CIDetector detectorOfType:context:options:]

/* Specifies a detector type for face recognition. */
CORE_IMAGE_EXPORT NSString* const CIDetectorTypeFace NS_AVAILABLE(10_7, 5_0);

/* Specifies a detector type for rectangle detection. */
CORE_IMAGE_EXPORT NSString* const CIDetectorTypeRectangle NS_AVAILABLE(10_10, 8_0);

/* Specifies a detector type for barcode detection. */
CORE_IMAGE_EXPORT NSString* const CIDetectorTypeQRCode NS_AVAILABLE(10_10, 8_0);

/* Specifies a detector type for text detection. */
#if __OBJC2__
CORE_IMAGE_EXPORT NSString* const CIDetectorTypeText NS_AVAILABLE(10_11, 9_0);
#endif

具体实现代码:

    // 初始化 - type选择CIDetectorTypeRectangle
    CIDetector *detector = [CIDetector detectorOfType:CIDetectorTypeRectangle context:nil options:@{CIDetectorAccuracy : CIDetectorAccuracyHigh}];
    
    // 获取矩形区域数组
    NSArray <CIFeature *>*rectangles = [detector featuresInImage:[CIImage imageWithCGImage:image.CGImage]];
    
    if (rectangles.count == 0) {
        
        return [[CIRectangleFeature alloc] init];
    } else {
        
        // 最大矩形区域
        CIRectangleFeature *rectangle = (CIRectangleFeature *)rectangles.firstObject;
        
        CGFloat rectangleRect = 0;
        
        for (CIRectangleFeature *rect in rectangles) {
            
            CGPoint p1 = rect.topLeft;
            CGPoint p2 = rect.topRight;
            CGFloat width = hypotf(p1.x - p2.x, p1.y - p2.y);
            
            CGPoint p3 = rect.topLeft;
            CGPoint p4 = rect.bottomLeft;
            CGFloat height = hypotf(p3.x - p4.x, p3.y - p4.y);
            
            CGFloat currentRectangleRect = height + width;
            
            // 获取最大矩形rect
            if (rectangleRect < currentRectangleRect) {
                
                rectangleRect = currentRectangleRect;
                rectangle = rect;
            }
        }
        return rectangle;
    }
CIRectangleFeature包含五个属性
    // 矩形区域
    CGRect bounds;
    // 矩形的四个顶点
    CGPoint topLeft;
    CGPoint topRight;
    CGPoint bottomLeft;
    CGPoint bottomRight;

通过CIRectangleFeature的四个顶点可以进行矩形区域的显示和裁剪

2、获取到矩形区域四个顶点使用LImageView显示矩形区域,并可以进行自由裁剪
/// 通过图片中矩形的四个顶点设置选中框的8个点
/// @param points 四个顶点数组 - 顺序为:左上 -> 右上 -> 右下 -> 左下
- (void)setPointWithRectanglePoints:(NSArray *)points {
    
    self.areaView.hidden = NO;
    
    CGPoint topLeft = CGPointFromString(points[0]);
    CGPoint topRight = CGPointFromString(points[1]);
    CGPoint bottomRight = CGPointFromString(points[2]);
    CGPoint bottomLeft = CGPointFromString(points[3]);
    
    // 四个顶点都为0时 - 默认选中整个图片
    if (CGPointEqualToPoint(topLeft, CGPointZero) &&
        CGPointEqualToPoint(topRight, CGPointZero) &&
        CGPointEqualToPoint(bottomLeft, CGPointZero) &&
        CGPointEqualToPoint(bottomRight, CGPointZero)) {
        
        [self resetAreaViewRect];
        
        return;
    }
    
    /// 将图像坐标系转为UIKit坐标系
    
    // 图片与imageView宽高比
    CGFloat scaleW = CGRectGetWidth(self.imageView.frame)/self.originImage.size.width;
    CGFloat scaleH = CGRectGetHeight(self.imageView.frame)/self.originImage.size.height;
    
    CGAffineTransform transform = CGAffineTransformMakeTranslation(0.f, CGRectGetHeight(self.imageView.frame));
    transform = CGAffineTransformScale(transform, 1, -1);
    
    transform = CGAffineTransformScale(transform, scaleW, scaleH);
    
    /// 设置选中区域
    // 设置四个顶点位置
    self.topLeftAreaView.center = CGPointApplyAffineTransform(topLeft, transform);
    self.topRightAreaView.center = CGPointApplyAffineTransform(topRight, transform);
    self.bottomLeftAreaView.center = CGPointApplyAffineTransform(bottomLeft, transform);
    self.bottomRightAreaView.center = CGPointApplyAffineTransform(bottomRight, transform);
    
    // 设置四个中心点位置
    for (UIView *view in self.areaView.subviews) {
        
        if (view == self.topCenterAreaView ||
            view == self.leftCenterAreaView ||
            view == self.bottomCenterAreaView ||
            view == self.rightCenterAreaView) {
            
            AreaView *centerView = (AreaView *)view;
            
            centerView.center = CGPointMake(fabs(centerView.relationView1.center.x - centerView.relationView2.center.x)/2 + MIN(centerView.relationView1.center.x, centerView.relationView2.center.x), fabs(centerView.relationView1.center.y - centerView.relationView2.center.y)/2 + MIN(centerView.relationView1.center.y, centerView.relationView2.center.y));
        }
    }
    
    // 设置蒙层路径
    [self resetMaskPath];
}

LImageView中需要自己定义并实现该方法

3、使用CIImage- (CIImage *)imageByApplyingFilter:(NSString *)filterName withInputParameters:(nullable NSDictionary<NSString *,id> *)params方法对图片进行裁剪(可自动矫正图片)

\color{red}{注意:该方法中使用的顶点为图片坐标系的顶点}

/// 通过图片矩形的四个顶点截取图片 - 带矫正
/// @param image 图片
/// @param rectangleFeature 顶点
+ (UIImage *)clipRectangleImage:(UIImage *)image rectanglePoint:(CIRectangleFeature *)rectangleFeature {
    
    UIImage *img;

    /// 判断是否存在矩形框 - 防止截取到空白图片
    if (CGPointEqualToPoint(rectangleFeature.topLeft, CGPointZero) &&
        CGPointEqualToPoint(rectangleFeature.topRight, CGPointZero) &&
        CGPointEqualToPoint(rectangleFeature.bottomLeft, CGPointZero) &&
        CGPointEqualToPoint(rectangleFeature.bottomRight, CGPointZero)) {
        
        img = image;
    } else {
        
        NSMutableDictionary *rectangleCoordinates = [NSMutableDictionary dictionary];
        rectangleCoordinates[@"inputTopLeft"] = [CIVector vectorWithCGPoint:rectangleFeature.topLeft];
        rectangleCoordinates[@"inputTopRight"] = [CIVector vectorWithCGPoint:rectangleFeature.topRight];
        rectangleCoordinates[@"inputBottomLeft"] = [CIVector vectorWithCGPoint:rectangleFeature.bottomLeft];
        rectangleCoordinates[@"inputBottomRight"] = [CIVector vectorWithCGPoint:rectangleFeature.bottomRight];

        CIImage * ciimage = [[CIImage imageWithCGImage:image.CGImage] imageByApplyingFilter:@"CIPerspectiveCorrection" withInputParameters:rectangleCoordinates];
        
        img = [UIImage imageWithCIImage:ciimage];
    }
    
    return img;
}
若使用图片裁剪工具进行图片裁剪需要先将坐标系转为图片坐标系,并将上边图片裁剪方法中的参数CIRectangleFeature改为自己定义的结构体或其他类型参数即可,方法为:
   // 将UIKit坐标系转换为图片坐标系
    
    CGFloat scaleW = self.originImage.size.width/CGRectGetWidth(self.imageView.frame);
    CGFloat scaleH = self.originImage.size.height/CGRectGetHeight(self.imageView.frame);
    
    CGAffineTransform transform = CGAffineTransformMakeTranslation(0.f, self.originImage.size.height);
    
    transform = CGAffineTransformScale(transform, scaleW, scaleH);
    
    transform = CGAffineTransformScale(transform, 1, -1);
    
    
    RectanglePoints points;
    
    points.topLeft = CGPointApplyAffineTransform(self.topLeftAreaView.center, transform);
    points.topRight = CGPointApplyAffineTransform(self.topRightAreaView.center, transform);
    points.bottomRight = CGPointApplyAffineTransform(self.bottomRightAreaView.center, transform);
    points.bottomLeft = CGPointApplyAffineTransform(self.bottomLeftAreaView.center, transform);

RectanglePoints为自己定义的结构体,包含四个顶点坐标

文笔有限,可能有说的不太清楚的地方,敬请谅解。如有错误,请多指教!

Just do IT!

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念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