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
| """
| 开盘啦工具
| """
| import easyocr
|
| reader = easyocr.Reader(['ch_sim', 'en'], gpu=True, model_storage_directory="res/ocr_model/model",
| download_enabled=False)
|
|
| def recognize_datas(img):
| results = reader.readtext(img, detail=1)
| start_y = -1
| end_x = -1
| index = 0
| datas = []
| for result in results:
| text = result[1]
| if text.find("股票名称") > -1:
| start_y = result[0][0][1]
|
| if text.find("竞价涨幅") > -1:
| end_x = result[0][0][0]
| if start_y > 0 and end_x > 0:
| if result[0][0][0] < end_x and result[0][0][1] > start_y and (result[0][1][0] - result[0][0][0]) > 30:
| datas.append(text)
| index += 1
| datas = datas[:3 * 10]
| fdatas = []
| temp = []
| for data in datas:
| temp.append(data)
| if len(temp) == 3:
| fdatas.append((temp[2][:6], temp[1]))
| temp = []
| return fdatas
|
|
