如何解决如何知道图像属于哪一类?
我正在尝试使用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.ml模型。
解决方法
此模型检测1000个类别的对象。其中约有200个是犬种。找出哪些类别是狗,并且仅在检测到的类别是狗类别之一时才显示某些内容。
如果要检测其他类型的狗,则必须训练自己的模型。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。