Swift + Core ML从数组中查找相似图像

import CoreML
import Vision

// 加载训练好的图像识别模型
guard let model = try? VNCoreMLModel(for: YourImageRecognitionModel().model) else {
    fatalError("无法加载图像识别模型")
}

// 创建图像识别请求
let request = VNCoreMLRequest(model: model) { request, error in
    guard let results = request.results as? [VNClassificationObservation],
          let topResult = results.first else {
        fatalError("无法识别图像")
    }
    
    // 输出最相关的图像识别结果
    print("相似图像识别结果：\(topResult.identifier)")
}

// 加载要识别的图像
guard let image = UIImage(named: "your_image.jpg"),
      let ciImage = CIImage(image: image) else {
    fatalError("无法加载图像")
}

// 创建图像处理请求
let imageRequestHandler = VNImageRequestHandler(ciImage: ciImage)

// 发送图像处理请求
do {
    try imageRequestHandler.perform([request])
} catch {
    fatalError("图像处理请求失败：\(error.localizedDescription)")
}