tawondazz/Mobile-Agent
0
1o
2 ��tf� � @ sH d dl Z d dlZd dlmZmZ d dlmZ dd� Zdd� Z dd � Z3dS )4� N)�5crop_image�calculate_size)�Imagec C s� t �| ��ddg�}t �|d�}||jd }t �|d d �df |d |d d �df |d �}|t �|� }|�ddg�}|d d |d krXt �|dd � |d d� g�}|�ddg��d�}|S )N� � r � ������ Zfloat32) �np�array�reshape�sum�shapeZarctan2ZargsortZ concatenateZastype)Zcoor�arrZsum_�centroid�thetaZ sort_points� r �?e:\Desktop\Mobile-Agent-v2-api\MobileAgent\text_localization.py� order_point s 4 r c s� t | �}t |�� � fdd�t|d �D �}td|d �D ]>}td� d �D ]4}| |d ||d krD||d |d d || |< q%t||d | || |d �|| |< q%q || � S )Nc s g | ] }d g� d �qS )r r r )�.0�_��nr r �6<listcomp> s z3longest_common_substring_length.<locals>.<listcomp>r )�len�range�max)Zstr1Zstr2�mZdp�i�jr r r � longest_common_substring_length s ",� r c C s� g }g }t �| �}||�}|d }t|jd �D ]A}t|| �}t||�} z7|| �d d }8W n Y qdd� t|�d��D �} | d | d | d | d g} |�|9� |�| � q||fS )10NZpolygonsr �textc S s g | ]}t |��qS r )�int)r �er r r r 3 s zocr.<locals>.<listcomp>r r r � ) �cv2Zimreadr r r r �listr �append) �11image_path�
ocr_detection�ocr_recognitionZ text_data�12coordinateZ13image_fullZ14det_resultr ZptsZ15image_crop�result�boxr r r �ocr# s" 16 17 18 r. ) r% �numpyr19 ZMobileAgent.cropr r �PILr r r r. r r r r �<module> s