开源ocr 识别手机截图定位
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | # coding: utf-8# import uiautomator2 as u2 import time from tqdm import tqdm from loguru import logger from cnocr import CnOcr def calculate_centroid(points): if len (points) = = 0 : return None num_points = len (points) sum_x = sum (point[ 0 ] for point in points) sum_y = sum (point[ 1 ] for point in points) centroid_x = sum_x / num_points centroid_y = sum_y / num_points print (pos_format(centroid_x,centroid_y)) return (centroid_x, centroid_y) # Example usage def pos_format(x,y): pos_w = "{:.4f}" . format (x / width) pos_y = "{:.4f}" . format (y / hight) return (pos_w,pos_y) if __name__ = = '__main__' : d = u2.connect( "5f83933a" ) image = d.screenshot() image.save( "home.jpg" ) width = image.width hight = image.height img_fp = 'home.jpg' ocr = CnOcr() # 所有参数都使用默认值 out = ocr.ocr(image) for line in out: print (f "{calculate_centroid(line['position'])} --- {line['text']} --- {line['score']}" ) print (out) |