Skip to content

Instantly share code, notes, and snippets.

@anupamchugh
Created October 15, 2019 06:18
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save anupamchugh/aa06e98554c83f30c7566167d2e48a92 to your computer and use it in GitHub Desktop.
var vnTextDetectionRequest : VNDetectTextRectanglesRequest{
let request = VNDetectTextRectanglesRequest { (request,error) in
if let error = error as NSError? {
print("Error in detecting - \(error)")
return
}
else {
guard let observations = request.results as? [VNTextObservation]
else {
return
}
var numberOfWords = 0
for textObservation in observations {
var numberOfCharacters = 0
for rectangleObservation in textObservation.characterBoxes! {
let croppedImage = crop(image: self.currentImage, rectangle: rectangleObservation)
if let croppedImage = croppedImage {
let processedImage = preProcess(image: croppedImage)
self.imageClassifier(image: processedImage,
wordNumber: numberOfWords,
characterNumber: numberOfCharacters, currentObservation: textObservation)
numberOfCharacters += 1
}
}
numberOfWords += 1
}
DispatchQueue.main.asyncAfter(deadline: .now() + 3, execute: {
self.drawRectanglesOnObservations(observations: observations)
})
}
}
request.reportCharacterBoxes = true
return request
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment