kumarAnurag/ocr_image_file_processing
1
1a
2 ���fa � @ s4 d dl Z d dlZd dlZdd� Zdd� Zdd� ZdS )� Nc C s t �|| � d S )N)�cv2Zimwrite)�image�filename� r �YC:\Users\kumar\OneDrive\Desktop\Hugginface_space\ocr_image_file_processing\ocr_process.py�save_debug_image s r c C sF t �| t j�}t jddd�}|�|�}t �|ddt jt j �\}}|S )Ng @)� r )Z clipLimitZ tileGridSizer �� )r ZcvtColorZCOLOR_BGR2GRAYZ createCLAHE�apply� thresholdZTHRESH_BINARY_INVZ THRESH_OTSU)�imgZgray_imgZclaheZ enhanced_img�_Z3binary_imgr r r �!preprocess_image_for_segmentation4 s5 6 r c C s: t j| d ddt jd�} t| �}t|d� tj|dd�}|S )N� )�fx�fy�
interpolationzfull_preprocessed_image.jpgzeng+hin)�lang)r �resizeZ INTER_CUBICr r � pytesseract�image_to_string)r Zpreprocessed_img�extracted_textr r r �extract_text_from_image s7 8r )r r �numpy�npr r r r r r r �<module> s9