我尝试了最近在iOS上推出的Core ML来识别和分类图像。问题是,它没有为发送的图像提供正确的结果。
如果我发送一个地球图像(地球仪),它会将其归类为气泡。以下是我使用的代码:
如果我发送一个地球图像(地球仪),它会将其归类为气泡。以下是我使用的代码:
let model = Resnet50()
let pixelBuffer: CVPixelBuffer = (modelImg.pixelBuffer())!
if let prediction = try? model.prediction(image: pixelBuffer) {
print( "Found it!! It is a/an \(prediction.classLabel)")
}
是否可以在现有的Core ML模型(例如Resnet50)基础上,使用自定义用例来训练模型?