Swift 在imagePickerController中获取所选图像并将其传递给coreML

我正在尝试使用coreML进行识别，该函数正常工作并正确显示结果。但是我想把这个方法调用成一个按钮，比如当我按下catDog按钮，它就会运行这个方法。但是因为finalResult（）和identificationCatOrDog（）是它自己的函数，所以我不能在按钮中调用它。我试图复制并粘贴按钮内的方法，但它没有显示任何内容。如何编辑代码，使findResult（）仅在我按下按钮时工作，而不是自动运行

import UIKit
import CoreML
import Vision
import Photos

class ViewController: UIViewController, UIImagePickerControllerDelegate, UINavigationControllerDelegate {

@IBOutlet var loadImage: UIImageView!
@IBOutlet var Result: UILabel!

@IBAction func photoBtn(_ sender: UIButton) {
    getPhoto()
}

@IBAction func cameraBtn(_ sender: UIButton) {
}

@IBAction func catDog(_ sender: UIButton) {
}

override func viewDidLoad() {
    super.viewDidLoad()
    // Do any additional setup after loading the view, typically from a nib.
}

override func didReceiveMemoryWarning() {
    super.didReceiveMemoryWarning()
    // Dispose of any resources that can be recreated.
}

func getPhoto() {
    let picker = UIImagePickerController()
    picker.delegate = self
    picker.sourceType = .photoLibrary
    present(picker, animated: true, completion: nil)
}

func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [UIImagePickerController.InfoKey : Any]) {
    picker.dismiss(animated: true, completion: nil)
    guard let gotImage = info[.originalImage] as? UIImage else {
        fatalError("No picture chosen")
    }

    loadImage.image = gotImage
    identifyCatOrDog(image: gotImage)
}

func identifyCatOrDog(image: UIImage) {
    let modelFile = ImageClassifier()
    let model = try! VNCoreMLModel(for: modelFile.model)

    let handler = VNImageRequestHandler(cgImage: image.cgImage!, options: [ : ])

    let request = VNCoreMLRequest(model: model, completionHandler: findResults)
    try! handler.perform([request])
}

func findResults(request: VNRequest, error: Error?) {
    guard let results = request.results as? [VNClassificationObservation] else {
        fatalError("Unable to get results")
    }

    var bestGuess = ""
    var bestConfidence: VNConfidence = 0

    for classification in results {
        if (classification.confidence > bestConfidence) {
            bestConfidence = classification.confidence
            bestGuess = classification.identifier
        }
    }

    Result.text = "Image is: \(bestGuess) with confidence \(bestConfidence) out of 1"
}

---

我认为问题在于，有时当图像选择器被解除时，您希望调用

identificationcatorlog

，但有时您不想

一个相当粗糙的可能性是：在按钮操作方法中，提高bool实例属性标志，以便在调用

didFinishPickingMedia

时，您知道是否调用

identificatorlog

---

一种更复杂的方法是将事物划分为助手类，以便按下

catDog

按钮后图像选择器的操作在一个完全不同的代码世界中进行。

我尝试使用UIswitch来解决这个问题，我添加了这些代码链接：if（catDogSwitch.isOn）{identificationcatordog（image:gotImage）}如果（faceSwitch.isOn）{identificationfaceswithlandmarks（image:gotImage）}