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