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

Question

我正在尝试使用coreML进行识别，该函数正常工作并正确显示结果。但是我想把这个方法调成一个按钮，就像我按下catDog按钮并运行方法一样。但是由于finalResult（）和identifyCatOrDog（）是它自己的函数，所以我不能把它调成按钮。我试图将方法复制并粘贴到按钮内，但它没有显示任何内容。如何编辑代码，以便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"
}

Answer 1

我认为问题在于，有时当图像选择器被解雇时，你想要调用identifyCatOrDog，但有时候你不需要。

一个相当粗略的可能性是：在按钮操作方法中，引发bool实例属性标志，以便在调用didFinishPickingMedia时知道是否调用identifyCatOrDog。

更复杂的方法是将事物分成辅助类，以便在按下catDog按钮后，图像选择器的操作发生在完全不同的代码世界中。