python利用OCR实现图片自动分类
off999 2024-12-03 00:08 28 浏览 0 评论
还是基于医院检查单的需求,后期会有大量的样本图片,需要分别归类去识别验证,通过程序自动分类减少下人工分类的成本,简单代码实现下。
思路
先通过百度普通OCR识别图片文字信息,对文字信息遍历,看是否有关联的关键字信息。如果识别不行,在本地的tesseract在识别一次,识别不出在统一放一个目录中,人工处理。
分类前:
分类后:
代码实现
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2022-7-8 上午 9:09
# @Author : damon.guo
# @File : picToClass.py
# @Software: PyCharm
import cv2
from PIL import Image
import pytesseract
import os,sys
import shutil
from aip import AipOcr
gettaglist = ["报告单", "报告", "医院", "卫生院", "中心"] # 识别关键字
def picToClass(picturePath):
# imageCode = cv2.imread(r"F:\xbl\11\\12.jpg") # 图像增强,二值化
# picturePath=r"F:\xbl\11\\preclass\\1.jpg" # 图像增强,二值化
image = cv2.imread(picturePath)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray, (5, 5), 0)
edged = cv2.Canny(gray, 75, 200)
tmpfilename = r"F:\xbl\11\\class\\tmp.jpg" # 临时文件
cv2.imwrite(tmpfilename, edged)
text = pytesseract.image_to_string(Image.open(tmpfilename), lang='chi_sim')
print("二次识别",picturePath)
for i in text.split("\n"):
index = getTagIndex(i)
if index != -1:
news = i[:index]
print("nes", news)
return news.strip(" ").strip(":").strip("%")
# 还是识别不出来,统一放一个目录
return "other"
def getclient(APP_ID, API_KEY, SECRET_KEY):
""" 你的 APPID AK SK """
# APP_ID = '你的 App ID'
# API_KEY = '你的 Api Key'
# SECRET_KEY = '你的 Secret Key'
client = AipOcr(APP_ID, API_KEY, SECRET_KEY)
return client
def get_file_content(imagepath):
with open(imagepath, 'rb') as fp:
return fp.read()
def ocrsdk(imagepath):
#先使用百度sdk识别
client = getclient("xx", "xx", "xx")
image = get_file_content(imagepath)
# 必填参数
options = {}
# options["templateSign"] = "" # 模板id
# options["detect_direction"] = "true" #是否自动转向
# options["probability"] = "true" #
options["language_type"] = "CHN_ENG" # 识别语言
res_image = client.basicAccurate(image,options)
print("识别图片:",image)
tag = baiduocrreslut(res_image)
if tag == "other":
# 无法识别在
# 在本地ocr识别一次
tag = picToClass(imagepath)
return tag
def getTagIndex(strs):
# 识别关键字在位置,取到索引,并向后多取2位
for i in gettaglist:
if i in strs:
index = strs.index(i)
return index+2
return -1
def copyFile(src,dst):
shutil.copy(src, dst)
def baiduocrreslut(res):
# 百度识别结果解析,
wordlist = res["words_result"]
for w in wordlist:
for i in gettaglist:
if i in w["words"]:
print(w["words"])
if len(w["words"]) >= 4:
return w['words']
return "other"
def main():
path = r"F:\xbl\preclass" # 待分类图片目录
classpath = r"F:\xbl\class" # 分类后主目录
listf = os.listdir(path)
for i in listf:
imagepath = os.path.join(path,i)
tag = ocrsdk(imagepath)
tagpath = os.path.join(classpath,tag)
if not os.path.exists(tagpath):
os.makedirs(tagpath)
newimagepath = os.path.join(tagpath,i)
if not os.path.exists(newimagepath):
shutil.copy(imagepath,newimagepath)
if __name__ == "__main__":
main()相关推荐
- 语音输入法哪个好用(语音输入法免费版下载安装)
-
日常基本使用的讯飞和搜狗都还不错,对比来看的话,讯飞相对会更好一些。讯飞输入法在键盘输入方面进步非常快,功能也比较全面。单纯的语音方面,讯飞的稍微好一点,例如一些标点、断句之类处理的更加到位。讯飞支持...
- 手机app应用市场(手机里面的应用市场)
-
360手机助手是android智能手机的手机管理专家。提供海量的游戏、软件、音乐、小说、视频、图片,通过它轻松下载、安装、管理手机资源。所有提供信息资源,全部经过360安全检测中心的审核认证,绿色无毒...
-
- 开局最强系统(开局最强系统txt奇书网)
-
《最强奶爸系统》作者:最美猪猪小说简介同学甲:“我找到了月入七千的工作。”同学乙:“我找到了月入一万的工作。叶辰,你呢?”叶辰:“我老婆…是万亿市值超凡集团的老板,给我生了10个孩子。”同学们:???叶辰遭前女友陷害,阴差阳错邂逅了一位美女...
-
2026-01-03 07:03 off999
- win10系统下载器官网(下载windows10官方)
-
如果您想要在Windows10上下载游戏辅助,可以按照以下步骤进行:1.打开您选择的游戏辅助下载网站,例如:GameCopyWorld或MegaGames。这些网站提供许多破解游戏的补丁和辅助工具...
- photoshop官方网站进入(photoshop官网入口)
-
那是因为网站采用了安全加密技术,安装了SSL证书,说明此网站为安全的页面,其信息已经经过加密措施。显示https的网站,网址上还会有显示一个小锁头,也是安全标志。1.先用Photoshop做一个网页图...
- windows7旗舰版主机多少钱(windows7旗舰版电脑多少钱一台)
-
其实这两个版本没有太大的差别,只是功能上有少量差别,旗舰版服务开启的要比专业版本服务开启的多几个,至于朋友们说的专业快速度比旗舰版快,或旗舰版比专业版速度快,个人认为速度都是一样只是功能上的差别,如果...
- 小米平板(小米平板怎么分屏)
-
小米平板的简单使用方法:1、首先给小米平板给充电,确保电量充足。2、长按住电源键启动小米平板,进入设备初始设置界面。3、根据屏幕上的提示信息,设置平板电脑联网、绑定小米账号等,直到完成即可正常使用了。...
- 移动硬盘一直在加载无法读取
-
移动硬盘电源灯亮说明硬盘有供电,无法识别此移动硬盘时,可按如下步骤进行排查:1)将移动硬盘拔下后,更换硬盘盒再接上确认。2)更换过硬盘盒还是无法识别的话,移动硬盘就损坏了,需要进行更换;3)若更换过硬...
- 下载软件的app大全(下载软件的app大全免费)
-
中国最常见的目前是迅雷下载软件。官网可以下载到正常的免费版本。他支持ftphttpbt磁力链接等多种形式的下载。国外的有bitcome。电驴和电骡等多种形式的下载软件。他们都可以满足日常的下载要求,...
- wifi贴小程序搭建需要多少钱
-
回答如下:要搭建自己的WiFi小程序,需要以下步骤:1.获得开发资格:首先需要注册成为开发者,获得小程序的开发资格。2.编写代码:使用微信小程序的开发工具,编写WiFi小程序的前端和后端代码。3....
- 外网服务器地址(外网服务器地址ip大全)
-
要弄懂这个问题,你首先要了解什么叫内网,什么叫外网,什么叫服务器,服务器在网络中所扮演的是什么角色,内网:就是指内部网络,窄义上的内网就是指中小型的局域网外网:就是指在你办公网络之外能访问到的网络...
- 360卸载不了的软件怎么办(用360卸载软件卸载不了)
-
开启了自我保护,关闭即可。解决方法如下:准备材料:360安全卫士、电脑1、在电脑上打开安全卫士,进去之后,点击右上方的列表图标,选择设置,2、进去安全卫士设置界面之后,点击安全防护中心,3、进去安全防...
-
- 联想电脑u盘启动不了(联想电脑为什么u盘启动不了)
-
1,启动计算机,并按住DEL键不放,但是也有按其它键的,所以请根据提示,直到出现BIOS设置窗口。2,键盘方向键向右移动到BIOS"Startup"菜单,然后选择“CMS”按回车键,然后选择"CMS"里面的Enabled。3,在对上面的操作...
-
2026-01-03 02:03 off999
- 海马苹果助手官方下载(海马苹果助手官方正版)
-
除了官方版,苹果其他服务器互通吧好像,只是不能换服务器登录你好;根据你的提示,可能是你的手机系统不正常蚂蚁浏览器,你可以现在一个手机QQ浏览器,它的体积小,功能强大,还特别的节省流量,它使用自主研发的...
- wifi恢复出厂设置后上不了网
-
恢复出厂设置后,路由器的网络设置也被清空了,需要重新设置账号和密码等信息并配置网络。还需要确认路由器的网线连接是否正确,以及是否有其他网络设备干扰。 如果重新配置还是连不上网不能用,可能需要...
欢迎 你 发表评论:
- 一周热门
-
-
抖音上好看的小姐姐,Python给你都下载了
-
全网最简单易懂!495页Python漫画教程,高清PDF版免费下载
-
Python 3.14 的 UUIDv6/v7/v8 上新,别再用 uuid4 () 啦!
-
飞牛NAS部署TVGate Docker项目,实现内网一键转发、代理、jx
-
python入门到脱坑 输入与输出—str()函数
-
宝塔面板如何添加免费waf防火墙?(宝塔面板开启https)
-
Python三目运算基础与进阶_python三目运算符判断三个变量
-
(新版)Python 分布式爬虫与 JS 逆向进阶实战吾爱分享
-
失业程序员复习python笔记——条件与循环
-
系统u盘安装(win11系统u盘安装)
-
- 最近发表
- 标签列表
-
- python计时 (73)
- python安装路径 (56)
- python类型转换 (93)
- python进度条 (67)
- python吧 (67)
- python的for循环 (65)
- python格式化字符串 (61)
- python静态方法 (57)
- python列表切片 (59)
- python面向对象编程 (60)
- python 代码加密 (65)
- python串口编程 (77)
- python封装 (57)
- python写入txt (66)
- python读取文件夹下所有文件 (59)
- python操作mysql数据库 (66)
- python获取列表的长度 (64)
- python接口 (63)
- python调用函数 (57)
- python多态 (60)
- python匿名函数 (59)
- python打印九九乘法表 (65)
- python赋值 (62)
- python异常 (69)
- python元祖 (57)
