Как проверить, нет ли текста в изображении - MLKit VisionCloudTextRecognizer - PullRequest
0 голосов
/ 24 июня 2019

У меня есть класс MLVision, который обрабатывает распознавание текста.

class MLVision {

    lazy var vision = Vision.vision()
    var textRecognizer: VisionTextRecognizer!
    var textArray = [] as Array<String>
    var lineText :String = ""

    func startRecognising(image : UIImage,completion: @escaping (_ result: [String]) -> ()){

        let visionImage = VisionImage(image: image)

        let options = VisionCloudTextRecognizerOptions()
        options.languageHints = ["en","hi"]
        textRecognizer = vision.cloudTextRecognizer(options: options)

        textRecognizer.process(visionImage, completion: { (features, error) in
            guard error == nil, let features = features else {
                print("No text Found!!")
                return
            }

            for block in features.blocks {
                for line in block.lines{


                    self.lineText = line.text
                    self.separate(lineText: self.lineText)

                }
            }
            completion(self.textArray)
        })
    }

    func separate(lineText : String){
        let separators = CharacterSet(charactersIn: (":)(,•/·]["))
        let ofWordsArray = lineText.components(separatedBy: separators)

        for word in ofWordsArray{

            let low = word.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()
            if low != ""{
                textArray.append(low)
            }
        }
    }
}

И это тот ViewController, который я называю классом Mlvision.

let mlVision = MLVision()

override func viewDidLoad() {
        super.viewDidLoad()
    mlVision.startRecognising(image: croppedImage!){ result in

                self.textArray = result
                self.searchIngredients(textArray: self.textArray)
        }
}

Таким образом, в основном мое приложение распознает текст, затем разделяет его и передает в массив.

Я хочу обработать, если я проверяю изображение без текста, чтобы показать предупреждение. Я знаю что guard error == nil, let features = features else {return} обрабатывает это, но я хочу проверить это из моего ViewController. Есть ли способ сделать это?

1 Ответ

2 голосов
/ 24 июня 2019

если я правильно понял ваш вопрос - вы просто хотите обрабатывать ошибки от распознавания в той области, где вы его запустили, если это так, чем вы можете изменить свой код ниже, используя Result .

// add your own Errors
enum RecognitionError: Error {
    case textNotFound
}

//modify `completion` closure by passing `Result` object instead of array
func startRecognising(image : UIImage,
                  completion: @escaping (_ result: Result<[String], RecognitionError>) -> ()){

    let visionImage = VisionImage(image: image)

    let options = VisionCloudTextRecognizerOptions()
    options.languageHints = ["en","hi"]
    textRecognizer = vision.cloudTextRecognizer(options: options)

    textRecognizer.process(visionImage, completion: { (features, error) in
        guard error == nil, let features = features else {
            print("No text Found!!")
            completion(.failure(.textNotFound))
            return
        }

        for block in features.blocks {
            for line in block.lines{


                self.lineText = line.text
                self.separate(lineText: self.lineText)

            }
        }
        completion(.success(self.textArray))
    })
}

И обработайте это в вашем ViewController:

override func viewDidLoad() {
    super.viewDidLoad()
    mlVision.startRecognising(image: croppedImage!){ result in
        switch result {

        case .success(let array):
            self.textArray = result
            self.searchIngredients(textArray: self.textArray)

        case .failure(let error):
            print(error)
        }
    }
}
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...