如何使用Swift、UIkit和CoreML在iOS应用程序中访问图像分类器ML模型的预测结果

import UIKit
import CoreML

class ViewController: UIViewController {
    // 创建图像分类器实例
    let imageClassifier = YourImageClassifierModel()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 加载和处理图像
        if let image = UIImage(named: "your_image.jpg") {
            if let pixelBuffer = image.pixelBuffer() {
                // 进行预测
                if let prediction = try? imageClassifier.prediction(input: pixelBuffer) {
                    // 获取预测结果
                    let predictedClass = prediction.classLabel
                    let confidence = prediction.classLabelProbs[predictedClass] ?? 0.0
                    
                    // 处理预测结果
                    print("预测结果：\(predictedClass)")
                    print("置信度：\(confidence)")
                }
            }
        }
    }
}

extension UIImage {
    // 将图像转换为模型所需的输入格式（CVPixelBuffer）
    func pixelBuffer() -> CVPixelBuffer? {
        let width = Int(self.size.width)
        let height = Int(self.size.height)
        
        var pixelBuffer: CVPixelBuffer?
        let status = CVPixelBufferCreate(kCFAllocatorDefault, width, height, kCVPixelFormatType_32ARGB, nil, &pixelBuffer)
        
        if status != kCVReturnSuccess {
            return nil
        }
        
        CVPixelBufferLockBaseAddress(pixelBuffer!, CVPixelBufferLockFlags(rawValue: 0))
        let pixelData = CVPixelBufferGetBaseAddress(pixelBuffer!)
        
        let colorSpace = CGColorSpaceCreateDeviceRGB()
        let context = CGContext(data: pixelData, width: width, height: height, bitsPerComponent: 8, bytesPerRow: CVPixelBufferGetBytesPerRow(pixelBuffer!), space: colorSpace, bitmapInfo: CGImageAlphaInfo.noneSkipFirst.rawValue)
        
        context?.translateBy(x: 0, y: CGFloat(height))
        context?.scaleBy(x: 1, y: -1)
        
        UIGraphicsPushContext(context!)
        self.draw(in: CGRect(x: 0, y: 0, width: width, height: height))
        UIGraphicsPopContext()
        
        CVPixelBufferUnlockBaseAddress(pixelBuffer!, CVPixelBufferLockFlags(rawValue: 0))
        
        return pixelBuffer
    }
}