如何知道图像属于哪个类？

Posted 2023-03-27

【中文标题】如何知道图像属于哪个类？

【英文标题】：How to know what class the image belongs to?

【发布时间】：2021-02-13 01:51:04

【问题描述】：

我正在尝试使用 Vision 和 CoreML。

代码：

extension CaptureImageView 
    private func loadImage() 
        guard let inputImage = image else  return 
        predictImage = inputImage
        performImageClassification()
    
    
    private func performImageClassification() 
        let image = self.predictImage
        let resizedImage = image?.resizeTo(size: CGSize(width: 224, height: 224))
        let buffer = resizedImage?.toBuffer()
        let output = try? model.prediction(image: buffer!)
        if let output = output 
            breed = output.classLabel
        
    

我想：

看看这是不是照片中的狗 如果是狗，请显示品种 如果它不是狗，请表现出一些警惕。

问题： 目前这种识别可以识别一切 :) 但我只想识别狗。

如前所述，我正在使用 Apple 的 Vision、CoreML、MobileNetV2.mlmodel。

【参考方案1】：

此模型可检测 1000 类对象。其中大约 200 种是犬种。找出哪些类是狗，并且仅在检测到的类是狗类之一时才显示一些东西。

如果要检测其他类型的狗，则必须训练自己的模型。