swift-CoreML-猫狗识别

一：我们需要在Main.storyboard中拉出来label和button和imageView

如下图所示 ：

1：在红框的位置选取button和label和imageView

image

2：之后将imageView拖拉进去，如下图所示

image

3：接下来，我们按照上面的方法来把button和label也放到imageView的上面，最终的效果图如下：

image

4：然后我们为了适配大多数的iphone机型，我们来给button添加限制条件

我们先给imageView添加限制条件吧！

在左下角的第二个按钮我们限制图片的width和height分别为374和816，距离手机上部距离为40，下部距离为6，左右距离为20

image

5：我们给label和button也加上这样的限制

image image

6:为了能够直观地看到button和label我们来改变这两个控件的颜色

在background中设置它的属性

image

7：完成之后，我们点击那个双圆圈按钮切换到双页面，按住control键拖拉storyboard中的控件到代码里面，并且设置按钮的名字。

image image

完成了storyboard之后，在代码端是下面这个样式

image

二：接下来我们的操作

（1）：我们把训练好的模型导入到我们的project中（拖拉到页面右端）

image

（2）：接下来我们给两个button添加点击事件

//当点击拍照时调用takephoto这个方法
camer_button.addTarget(self, action: #selector(takephoto), for: .touchUpInside)
//当点击从相册中选择的时候调用chooseImageFromLibrary这个方法
photos_button.addTarget(self, action: #selector(chooseImageFromLibrary), for: .touchUpInside)

（4）：接着我们在chooseImageFromLibrary中实现从相册中获取图片的功能

@objc func chooseImageFromLibrary(_ sender: Any) {
   //判断是否可以打开相册
       if UIImagePickerController.isSourceTypeAvailable(UIImagePickerController.SourceType.photoLibrary) {
           //调用打开相册权限的函数
           let imagePicker = UIImagePickerController()
           //设置代理
           imagePicker.delegate = self
           //设置控制器的类型---相册
           imagePicker.sourceType = UIImagePickerController.SourceType.photoLibrary
           //图片不允许编辑
           imagePicker.allowsEditing = false
           //跳转到控制器---相册
           self.present(imagePicker,animated: true,completion: nil)
       }
   }

三：我们选取图片之后，开始通过我们导入的模型来分析选中的图片,我们在imagePickerController这个函数中实现

//获取图片后的操作
func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [UIImagePickerController.InfoKey : Any]) {
        //获取图像
        if let pickedImage = info[UIImagePickerController.InfoKey.originalImage] as? UIImage {
            display_label.text = "分析中"
            //图片填充模式--等比例缩放图片
            image.contentMode = .scaleToFill
            image.layer.masksToBounds = true
            //在imageView上显示获取到的图片
            image.image = pickedImage
            //加载模型
            guard let model = try? VNCoreMLModel(for:ImageClassifier().model) else {
                fatalError("无法加载训练模型")
            }
            //创建一个模型请求
            let requset = VNCoreMLRequest(model: model) {
                [weak self] requesst,error in guard let results = requesst.results as? [VNClassificationObservation],
                    let topResult = results.first else {
                        fatalError("失败")
                }
                //主线程更新UI
                DispatchQueue.main.async {
                    print(topResult)
                    //显示识别的结果
                    self!.display_label.text = "识别的结果是：\(Int(topResult.confidence*100))%是\(topResult.identifier)"
                    //if topResult.confidence < 0.99{
                       // self!.display_label.text = "识别的结果是：\(Int(topResult.confidence*100))%是\(topResult.identifier)"
                    //}
                    //else {
                       // self!.display_label.text = "识别的结果是：非猫非狗"
                    }
                    
                }
            }
            //将获取到的图片转换为CIImage类型作为参数传递给模型
            guard let ciImage = CIImage(image: pickedImage) else {
                fatalError("不能获取选择到的图片")
            }
            //运行图片分类识别器
            let handler = VNImageRequestHandler(ciImage: ciImage)
            DispatchQueue.global().async {
                do {
                    try handler.perform([requset])
                }
                catch {
                    print(error)
                }
            }
            
        }
        picker.dismiss(animated: true, completion: nil)
}

最后，也是最重要的事情，需要在Info.plist中加入调用相机的权限

image