File size: 640 Bytes
bfabfea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from pipeline_functions import object_detection, crop_image, enhance_image, morphological_transform, hoffman_transform, pytesseract_rotate, ocr,ner
import os
def main(path):
img_name = os.path.basename(path)
seg_result, img_file = object_detection(path)
croped_img = crop_image(seg_result, img_file, img_name)
image = enhance_image(croped_img, img_name)
processed_img = morphological_transform(image)
rotated_image, image = hoffman_transform(processed_img, image)
img_name = pytesseract_rotate(rotated_image, image, img_name)
file_name = ocr(img_name)
Output_dict = ner(file_name)
return Output_dict |