首先到https://developer.apple.com/machine-learning/ 下载一个Resnet50.mlmodel文件,将它拖动到XCode项目文件夹:
编写代码如下:
import UIKit
import CoreML
class ViewController: UIViewController {
override func viewDidLoad() {
super.viewDidLoad()
// Do any additional setup after loading the view, typically from a nib.
let image = UIImage(named: "sample")
let width: CGFloat = 224.0
let height: CGFloat = 224.0
//获得一个基于位图的上下文,并设置其为当前的上下文
UIGraphicsBeginImageContext(CGSize(width: width, height: height))
//将从项目中加载的图像,绘制在上下文的指定区域
image?.draw(in: CGRect(x: 0, y: 0, width: width, height: height))
//从上下文中获得格式转换后的图像
let newImage = UIGraphicsGetImageFromCurrentImageContext()
//完成图像的格式转换后,关闭当前的上下文