Tag: coreml

将图像转换为CVPixelBuffer进行机器学习Swift

我正在尝试在2017年的WWDC上演示的苹果示例Core ML型号正常工作。 我正在使用GoogLeNet尝试对图像进行分类（请参阅Apple机器学习页面 ）。 该模型将CVPixelBuffer作为input。 我有一个名为imageSample.jpg的图片，我正在使用这个演示。 我的代码如下： var sample = UIImage(named: "imageSample")?.cgImage let bufferThree = getCVPixelBuffer(sample!) let model = GoogLeNetPlaces() guard let output = try? model.prediction(input: GoogLeNetPlacesInput.init(sceneImage: bufferThree!)) else { fatalError("Unexpected runtime error.") } print(output.sceneLabel) 我总是得到输出中意外的运行时错误，而不是图像分类。 我的代码转换图像如下： func getCVPixelBuffer(_ image: CGImage) -> CVPixelBuffer? { let imageWidth = Int(image.width) let imageHeight = Int(image.height) let attributes : […]