swift - CoreML 无法验证输入

Question

我对 CoreML 有疑问，因为当我希望程序验证图像时出现错误:

[coreml] Error Domain=com.apple.CoreML Code=1 "Input image feature image does not match model description" UserInfo={NSLocalizedDescription=Input image feature image does not match model description, NSUnderlyingError=0x2807c0cf0 {Error Domain=com.apple.CoreML Code=1 "Image is not expected type OneComponent8, instead is 32ARGB" UserInfo={NSLocalizedDescription=Image is not expected type OneComponent8, instead is 32ARGB}}} 2018-10-24 06:47:53.975118+0200 recognizeMyFood[25848:7075048] [coreml] Failure verifying inputs.

下面是我进行图像转换时截取的代码:

UIGraphicsBeginImageContextWithOptions(CGSize(width: 343, height: 447), true, 2.0)
image.draw(in: CGRect(x: 0, y: 0, width: 343, height: 447))
let newImage = UIGraphicsGetImageFromCurrentImageContext()!
UIGraphicsEndImageContext()

let attrs = [kCVPixelBufferCGImageCompatibilityKey: kCFBooleanTrue, kCVPixelBufferCGBitmapContextCompatibilityKey: kCFBooleanTrue] as CFDictionary
var pixelBuffer : CVPixelBuffer?
let status = CVPixelBufferCreate(kCFAllocatorDefault, Int(newImage.size.width), Int(newImage.size.height), kCVPixelFormatType_32ARGB, attrs, &pixelBuffer)
guard (status == kCVReturnSuccess) else {
    return
}

如果有人能帮助我，我将不胜感激!

Answer 1

关键在错误信息中:“Image is not expected type OneComponent8, instead is 32ARGB”

您为其提供彩色图像 (kCVPixelFormatType_32ARGB)，但模型需要灰度图像 (​​kCVPixelFormatType_OneComponent8)。