環(huán)境
- Xcode 9 Beta 及以上
過程
1.新建“Single View App”工程
1.png
2.去官網(wǎng)提供的模型資源中斑鸦，下載模型文件(我下載的第一個)：
2.jpeg
3.將下載后的“GoogLeNetPlaces.mlmodel”文件直接拖到工程中：
3.jpeg
4.工程對應的“Build Phases -> Compile Sources”中，添加上面拖進來的文件：
4.jpeg
5.在“Assets.xcassets”添加一張尺寸為“224*224”像素的png圖牍白，取名“test”：
5.jpeg
6.在“ViewContrller.m”中敲代碼：
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;
}
工程代碼下載地址：傳送門
