目录
1安装所需要的工具
1安装cuda,cudnn
CUDA Toolkit Archive | NVIDIA Developer
安装
我选的是Network下载
然后就是一路下一步
查看自己是否安装成功
cuDNN Archive | NVIDIA Developer
下载cudnn需要注册一个账号
登陆后下载
下载好后会有三个文件
找到Cuda的目录
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
把这三个文件复制进去
anaconda可以帮助我们下载python的库。
2下载安装anaconda
Anaconda | Anaconda Distributionhttps://www.anaconda.com/products/distribution
建议别装D盘
勾选好点击安装
一路Next
点击Finish安装完成
我使用的pycharm来连接anacond
3安装pycharm
Download PyCharm: Python IDE for Professional Developers by JetBrains
我这边安装的是pycharm2022.2.3
写好文件地址点击next
可以按照自己的需要点击,但是最好添加PATH
2开始搭建
1打开我们之前安装的anaconda
conda create -n pytorch python=3.9
创建一个名字为pytorch,python版本为3.9的一个虚拟环境
进入我们的虚拟环境我们安装包就可以从这里install
我们需要使用pycharm来连接anaconda
点击New project
点击Add interpreter来引入我们刚刚创建的虚拟环境
虚拟环境文件一般都在
anaconda/envs/
点击create
pycharm右下角可以看到当前使用的anaconda环境
2下一步安装pytorch,
torch,torchvision,torchaudio
官网有对应的表
Previous PyTorch Versions | PyTorch
按照表上的内容选择
我的cuda版本对应11.3
使用anaconda下载
成功下载
3测试
安装easyocr
www.jaided.ai模型库里下载模型
上代码测试
#@author=m0_74401015
import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
import re
import sys
import easyocr
import pyautogui
import pygame
import pyperclip
from pygame.locals import *
import string
def _getXY():#显示截取到的屏幕,使用两个点对其切割
pygame.init()#初始化
getScreen = pygame.display.Info()
screen=pygame.display.set_mode((getScreen.current_w,getScreen.current_h),pygame.NOFRAME)#定义窗口
bacground = pygame.image.load(r"windowsimg.png")#图片
i=0
list1=[]
# list2=[]
# clist=[list1+list2]
while True:
screen.blit(bacground,(0,0))
for m in pygame.event.get():
if m.type == MOUSEBUTTONDOWN: #获取鼠标动作
pos = pygame.mouse.get_pos()
# print(pos)
list1.append(pos)
i=i+1
if m.type == MOUSEWHEEL:
return 0 #仅退出此函数返回0
# sys.exit() #此命令导致图片未分析就结束程序
pygame.display.update()
if i == 2: #获取到两个坐标退出
img = pyautogui.screenshot(region=(list1[0][0], list1[0][1], list1[1][0] - list1[0][0], list1[1][1] - list1[0][1]))
# img = pyscreenshot.grab(bbox=(list1[0][0], list1[0][1], list1[1][0] - list1[0][0], list1[1][1] - list1[0][1]))
img.save('Cutpicture.png')
def get_window():#截取屏幕
img = pyautogui.screenshot(region=None)
img.save('windowsimg.png')
def img_SwitchScripts():#切割好的图片转文字并分析
i = []
reader = easyocr.Reader(['ch_sim','en'],model_storage_directory=r'./.EasyOCR/model')
readimg = reader.readtext(r'Cutpicture.png')
for r in readimg:
i.append(r)
# i+=r
l=tuple(i)
for li in l:
li=re.findall(r"['](.*?)[']",str(l))
lii="".join(li)
# print(text)
pyperclip.copy(lii)
# pyperclip.paste()
if __name__ == '__main__':#入口
get_window()
_getXY()
img_SwitchScripts()
成功识别