六张图迅速搭建CoreML入门工程(Object-C)

环境

  • Xcode 9 Beta 及以上

过程

1.新建“Single View App”工程

六张图迅速搭建CoreML入门工程(Object-C)_第1张图片
1.png

2.去官网提供的模型资源中,下载模型文件(我下载的第一个):

六张图迅速搭建CoreML入门工程(Object-C)_第2张图片
2.jpeg

3.将下载后的“GoogLeNetPlaces.mlmodel”文件直接拖到工程中:

六张图迅速搭建CoreML入门工程(Object-C)_第3张图片
3.jpeg

4.工程对应的“Build Phases -> Compile Sources”中,添加上面拖进来的文件:

六张图迅速搭建CoreML入门工程(Object-C)_第4张图片
4.jpeg

5.在“Assets.xcassets”添加一张尺寸为“224*224”像素的png图,取名“test”:

六张图迅速搭建CoreML入门工程(Object-C)_第5张图片
5.jpeg

6.在“ViewContrller.m”中敲代码:

六张图迅速搭建CoreML入门工程(Object-C)_第6张图片
6.jpeg

其中的代码段包括:

#import "GoogLeNetPlaces.h"
@property (nonatomic, strong) GoogLeNetPlaces *model;

// 初始化模型对象
self.model = [[GoogLeNetPlaces alloc] init];
// 初始化需要输入的图片信息
UIImage *image = [UIImage imageNamed:@"test"];
CGImageRef cgImage = image.CGImage;
// 得到输出
GoogLeNetPlacesOutput *outPut = [self.model predictionFromSceneImage:[self pixelBufferFromCGImage:cgImage] error:nil];
// 打印输出结果
NSLog(@"Dict:%@, \n label:%@", outPut.sceneLabelProbs, outPut.sceneLabel);
- (CVPixelBufferRef)pixelBufferFromCGImage:(CGImageRef)image{
    NSDictionary *options = [NSDictionary dictionaryWithObjectsAndKeys:
                             [NSNumber numberWithBool:YES], kCVPixelBufferCGImageCompatibilityKey,
                             [NSNumber numberWithBool:YES], kCVPixelBufferCGBitmapContextCompatibilityKey,
                             nil];
    CVPixelBufferRef pxbuffer = NULL;
    CGFloat frameWidth = CGImageGetWidth(image);
    CGFloat frameHeight = CGImageGetHeight(image);
    CVReturn status = CVPixelBufferCreate(kCFAllocatorDefault, frameWidth, frameHeight, kCVPixelFormatType_32ARGB, (__bridge CFDictionaryRef) options, &pxbuffer);
    NSParameterAssert(status == kCVReturnSuccess && pxbuffer != NULL);
    CVPixelBufferLockBaseAddress(pxbuffer, 0);
    void *pxdata = CVPixelBufferGetBaseAddress(pxbuffer);
    NSParameterAssert(pxdata != NULL);
    CGColorSpaceRef rgbColorSpace = CGColorSpaceCreateDeviceRGB();
    CGContextRef context = CGBitmapContextCreate(pxdata, frameWidth, frameHeight, 8, CVPixelBufferGetBytesPerRow(pxbuffer), rgbColorSpace, (CGBitmapInfo)kCGImageAlphaNoneSkipFirst);
    NSParameterAssert(context);
    CGContextConcatCTM(context, CGAffineTransformIdentity);
    CGContextDrawImage(context, CGRectMake(0, 0, frameWidth, frameHeight), image);
    CGColorSpaceRelease(rgbColorSpace);
    CGContextRelease(context);
    CVPixelBufferUnlockBaseAddress(pxbuffer, 0);
    return pxbuffer;
}

工程代码下载地址:传送门

你可能感兴趣的:(六张图迅速搭建CoreML入门工程(Object-C))