python3.5+tesseract+adb实现西瓜视频或头脑王者辅助答题,python3.5tesseract
python3.5+tesseract+adb实现西瓜视频或头脑王者辅助答题,python3.5tesseract
最近的答题赢钱很火爆,我也参与了几次,有些题目确实很难答,但是10秒钟的时间根本不够百度的,所以写了个辅助挂,这样可以出现题目时自动百度,这个时间也就花掉2秒钟,剩下的7、8秒钟可以进行分析和作答,提升了赢钱概率。
源码可以见我的github:点击链接
原理分析下:使用adb命令,抓取手机视频播放的界面,然后通过python的截取和ocr,获得到题目和答案, 然后百度得到结果。这个环境怎么搭建,有需要的童鞋可以联系我,因为使用本地的ocr所以解析不花钱,也没有使用的限制。
github上的代码中
ocr_bw.py,这个是自动根据题目去百度,然后打开浏览器,展示检索结果
# -*- coding: utf-8 -*- import pytesseract import time import webbrowser import subprocess from PIL import Image def main(): """ 主函数 """ op = yes_or_no('请确保手机打开了 ADB 并连接了电脑,' '然后打开西瓜视频后再用本程序,确定开始?') if not op: print('bye') return #核心递归 ocr_subject_parent() # for root, sub_dirs, files in os.walk('E:/临时接收的文件/知乎答题/百万/'): # for file in files: # print('发现图片:' + file) # img = Image.open('E:/临时接收的文件/知乎答题/百万/'+file) # ocr_subject(img) def yes_or_no(prompt, true_value='y', false_value='n', default=True): """ 检查是否已经为启动程序做好了准备 """ default_value = true_value if default else false_value prompt = '{} {}/{} [{}]: '.format(prompt, true_value, false_value, default_value) i = input(prompt) if not i: return default while True: if i == true_value: return True elif i == false_value: return False prompt = 'Please input {} or {}: '.format(true_value, false_value) i = input(prompt) def screenImg(true_value='', default=True): prompt = '当出现题目时,请按下回车进行识别 ' i = input(prompt) if not i: return default while True: if i == true_value: return True else: return False i = input(prompt) def ocr_subject(p): # 截取 距离上530开始 940结束 # 截取 距离上260 570结束 p = cut_img(p) pytesseract.pytesseract.tesseract_cmd = 'E:/Program Files (x86)/Tesseract-OCR/tesseract' subject = pytesseract.image_to_string(p, lang='chi_sim') subject = "".join(subject.split()) subject = subject.split('.')[1] print(subject) openPage(subject) ocr_subject_parent() def ocr_subject_parent(): result = screenImg() if result: start = time.time() # screenshot.check_screenshot() process = subprocess.Popen( 'adb shell screencap -p', shell=True, stdout=subprocess.PIPE) binary_screenshot = process.stdout.read() binary_screenshot = binary_screenshot.replace(b'\r\n', b'\n') f = open('autojump.png', 'wb') f.write(binary_screenshot) f.close() # screenshot.pull_screenshot() img = Image.open('autojump.png') print("耗时:" + str(time.time() - start)) ocr_subject(img) def openPage(subject): url = 'https://www.baidu.com/s?wd={}'.format( subject) webbrowser.open(url) webbrowser.get() def cut_img(img): region = img.crop((70, 260, 1025, 570)) #region.save("temp/cut_first.png") return region if __name__ == '__main__': main()
ocr_bw2.py,这个是根据题目+答案,去百度检索,通过爬虫抓取百度的收录数,然后在控制台打印结果
__author__ = 'zjy' # -*- coding:utf-8 -*- import pytesseract import time import webbrowser import subprocess from PIL import Image import urllib import urllib.request import threading from urllib.parse import quote def main(): """ 主函数 """ op = yes_or_no('请确保手机打开了 ADB 并连接了电脑,' '然后打开西瓜视频后再用本程序,确定开始?') if not op: print('bye') return # 核心递归 ocr_subject_parent() # for root, sub_dirs, files in os.walk('E:/临时接收的文件/知乎答题/百万/'): # for file in files: # print('发现图片:' + file) # img = Image.open('E:/临时接收的文件/知乎答题/百万/'+file) # ocr_subject(img) def yes_or_no(prompt, true_value='y', false_value='n', default=True): """ 检查是否已经为启动程序做好了准备 """ default_value = true_value if default else false_value prompt = '{} {}/{} [{}]: '.format(prompt, true_value, false_value, default_value) i = input(prompt) if not i: return default while True: if i == true_value: return True elif i == false_value: return False prompt = 'Please input {} or {}: '.format(true_value, false_value) i = input(prompt) def screenImg(true_value='', default=True): prompt = '当出现题目时,请按下回车进行识别 \n' i = input(prompt) if not i: return default while True: if i == true_value: return True else: return False i = input(prompt) def ocr_subject(p): # 截取 距离上530开始 940结束 # 截取 距离上260 570结束 subImg = cut_img(p) pytesseract.pytesseract.tesseract_cmd = 'E:/Program Files (x86)/Tesseract-OCR/tesseract' subject = pytesseract.image_to_string(subImg, lang='chi_sim') subject = "".join(subject.split()) subject = subject.split('.')[1].replace("\"", "") print(subject) ocr_answer(p, subject) # openPage(subject) # print("结束:" + str(time.time())) ocr_subject_parent() def getSearchNum(key): key = quote(key) # print(key) url = 'http://www.baidu.com/s?wd={}'.format(key) # print(url) response = urllib.request.urlopen(url) page = response.read().decode("utf-8") i = int(page.index('百度为您找到相关结果约')) start = i + 10 end = i + 25 page = page[start: end] return page def ocr_answer(p, subject): list = cut_question(p) pytesseract.pytesseract.tesseract_cmd = 'E:/Program Files (x86)/Tesseract-OCR/tesseract' for p in list: t = threading.Thread(target=ocr_answer_thread, args=(p, subject)) t.start() def ocr_answer_thread(p, subject): answer = pytesseract.image_to_string(p, lang='chi_sim') answer = "".join(answer.split()) v = getSearchNum(subject + ' ' + answer) print(answer + ' ' + v) # print(time.time()) def ocr_subject_parent(): result = screenImg() if result: start = time.time() # print("开始:" + str(start)) # screenshot.check_screenshot() process = subprocess.Popen( 'adb shell screencap -p', shell=True, stdout=subprocess.PIPE) binary_screenshot = process.stdout.read() binary_screenshot = binary_screenshot.replace(b'\r\n', b'\n') f = open('autojump.png', 'wb') f.write(binary_screenshot) f.close() # screenshot.pull_screenshot() img = Image.open('autojump.png') ocr_subject(img) def openPage(subject): url = 'https://www.baidu.com/s?wd={}'.format( subject) webbrowser.open(url) webbrowser.get() def cut_img(img): region = img.crop((70, 260, 1025, 570)) # region.save("temp/cut_first.png") return region def cut_question(img): list = [] question1 = img.crop((70, 590, 1025, 768)) question2 = img.crop((70, 769, 1025, 947)) question3 = img.crop((70, 948, 1025, 1130)) list.append(question1) list.append(question2) list.append(question3) # question1.save("temp/cut_1.png") # question2.save("temp/cut_2.png") # question3.save("temp/cut_3.png") return list if __name__ == '__main__': main()
由于很多题目是下列哪个不是,所以我更喜欢用第一个方式,基本上识别时间在0.5-0.6秒之间。
最后里面的ocr_zh.py是可以用来抓取头脑王者的辅助。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持帮客之家。
评论关闭