def recognize_text(self, img, task=""):
word_list = []
print("Image Processing Started")
img_handler = ImageHandler()
words = img_handler.split_text(img, task)
for word in words:
img = cv2.cvtColor(word, cv2.COLOR_BGR2GRAY)
# img = img_handler.preprocess_normal_handwriting(img)
img = img_handler.preprocess(img, self.img_size)
word_list.append(img)
# cv2.imshow('word', word)
# cv2.waitKey(0)
print("Image Processing Finished")
print("Recognizing Text Started")
batch = Batch(None, word_list)
recognized_list = self.model.batch_test(batch)
print('Image Text: ', recognized_list)
text = ''
for i in recognized_list:
text += i + ' '
return text
def test_address(self, img, task=""):
word_list = []
img_handler = ImageHandler()
words = img_handler.split_text(img, task)
for word in words:
img = cv2.cvtColor(word, cv2.COLOR_BGR2GRAY)
# img = img_handler.preprocess_normal_handwriting(img)
img = img_handler.preprocess(img, self.img_size)
word_list.append(img)
# cv2.imshow('word', word)
# cv2.waitKey(0)
batch = Batch(None, word_list)
recognized_list = self.model.batch_test(batch)
return recognized_list
