forked from nickliqian/cnn_captcha
-
Notifications
You must be signed in to change notification settings - Fork 0
/
recognize_local.py
58 lines (45 loc) · 1.67 KB
/
recognize_local.py
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#!/usr/bin/python
# -*- coding: UTF-8 -*-
"""
使用自建的接口识别来自网络的验证码
需要配置参数:
remote_url = "https://www.xxxxxxx.com/getImg" 验证码链接地址
rec_times = 1 识别的次数
"""
import datetime
import requests
from io import BytesIO
import time
import json
import os
def recognize_captcha(test_path, save_path, image_suffix):
image_file_name = 'captcha.{}'.format(image_suffix)
with open(test_path, "rb") as f:
content = f.read()
# 识别
s = time.time()
url = "http://127.0.0.1:6000/b"
files = {'image_file': (image_file_name, BytesIO(content), 'application')}
r = requests.post(url=url, files=files)
e = time.time()
# 识别结果
print("接口响应: {}".format(r.text))
predict_text = json.loads(r.text)["value"]
now_time = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
print("【{}】 耗时:{}ms 预测结果:{}".format(now_time, int((e-s)*1000), predict_text))
# 保存文件
img_name = "{}_{}.{}".format(predict_text, str(time.time()).replace(".", ""), image_suffix)
path = os.path.join(save_path, img_name)
with open(path, "wb") as f:
f.write(content)
print("============== end ==============")
def main():
with open("conf/sample_config.json", "r") as f:
sample_conf = json.load(f)
# 配置相关参数
test_path = "sample/test/0401_15440848576253345.png" # 测试识别的图片路径
save_path = sample_conf["local_image_dir"] # 保存的地址
image_suffix = sample_conf["image_suffix"] # 文件后缀
recognize_captcha(test_path, save_path, image_suffix)
if __name__ == '__main__':
main()