百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

python利用OCR实现图片自动分类

off999 2024-12-03 00:08 22 浏览 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()

相关推荐

装系统的u盘怎么制作(装系统怎么制作u盘启动盘)
装系统的u盘怎么制作(装系统怎么制作u盘启动盘)

1、下载安装u当家u盘启动盘制作工具,完成之后打开该软件2、将之前准备好的U盘插入电脑,U当家会自动识别并选为默认,如果有多个U盘可以在“选择U盘”的下拉框中选择要制作成U盘启动盘的U盘。3、点击下面的“一键制作”按钮。(如果是电脑主板是U...

2025-11-15 07:03 off999

台电u盘怎么样

感觉台电U盘还是可以的,我知道台电是国内最成功的数码产品制造商之一,是一个致力于IT及消费数码类产品研发、生产、销售及服务的一体化品牌,U盘很好用,用的人很多。威刚是480Mbit/S只是USB2...

u盘重装系统蓝屏进不去(u盘装系统重启蓝屏)

尊敬的用户您好:可按下面的步骤进行系统盘装系统:1.买一张win7光盘,将系统光盘放入光驱里。2.打开电源,然后观察屏幕的提示,在启动系统之前有按“f2进行bios设置。3.设置启动方式。找到boot...

win快捷键大全(win快捷键大全表)

win快捷键:单独按Windows:显示或隐藏“开始”功能表Windows+BREAK:显示“系统属性”对话框Windows+D:显示桌面或恢复桌面Windows+M:最小化所有窗口Windows...

u盘突然要格式化怎么回事(用着的u盘突然要格式化)

U盘提示格式化的原因可能有以下几点:文件系统损坏:U盘的文件系统可能出现损坏,导致无法正常读取和写入数据。病毒感染:U盘可能被病毒感染,病毒会破坏文件系统或隐藏文件。不兼容的文件系统:U盘可能使用了一...

路由器pppoe怎么设置(路由器如何设置pppoe上网)

移动宽带PPPOE是不用设置的,你只要把帐号和密码填上其他都不选就能上网。但是要注意连接到你电脑的网线是直接经过一个路由器后连接到猫上的,因为我见过有人是这样连接的猫---路由器---路由器---电脑...

win7桌面图标设置在哪里(win7桌面图标在哪个文件夹)

要调出Win7桌面图标,首先需要进入桌面。可以通过点击任务栏上方的“显示桌面”按钮或按下Windows键+D快捷键进入桌面。在桌面上右键单击空白区域,选择“个性化”,然后选择“更改桌面图标”选项。在弹...

win7手机壁纸(windows7手机桌面主题)

1.首先我们打开手机进入到桌面,找到设置图标点击打开。华为手机如何设置自定义锁屏壁纸2.进入到设置界面之后,点击下方的显示选项。华为手机如何设置自定义锁屏壁纸3.然后我们就可以在显示界面中找到壁纸...

怎么开启路由器wifi(怎么开启路由器的dhcp功能)

把路由器改为开放网络方法如下:1、打开浏览器,在地址栏输入路由器网关IP地址(路由器背后的标签上有,一般是192.168.1.1),输入登录用户名和密码(一般均为admin);2、登录成功后就显示运行...

win10加载不出来桌面黑屏(window10加载不出来)

方法一、如果是遇到了突发性黑屏首先尝试使用Ctrl+Alt+Delete组合键来进行电脑重启一般重启可以解决大部分黑屏故障。win10电脑黑屏按什么键恢复_win10电脑黑屏一键恢复的方法方法...

怎么用火绒关闭win安全中心(win10火绒安全软件怎么关闭)

一、火绒防火墙关闭方法在电脑上运行火绒软件后,点击首页界面的“防护中心”,在病毒防护界面可以关闭文件、U盘、邮件等保护功能,。系统防护:在该界面可以关闭系统防护功能,。网络防护:可以关闭网络保护等功...

微软应用商店下载手机版(微软应用商店在哪下)

、在桌面任务栏找到微软应用商店,点击并打开。2、进入页面后点击【应用】,进入应用页面。3、在页面里往下移动,找到热门应用,找到一个软件,点击进入。4、进入页面后,点击【安装】,软件会自动安装,安装完毕...

win11系统怎么查看电脑配置(win11怎么查看系统版本)

答:win11查看电脑配置步骤如下。1.点击下方任务栏的windows图标或者按下键盘“windows键”打开开始菜单。2.在开始菜单中找到“设置”或“settings”,一般是右上角的齿轮状按钮,3...

麦克风没坏但是没声音(麦克风没声怎么回事)

几种可能性,供您参考:1、麦是完全好的(其它机子上可以用)2、插孔没有插错3、音量控制里的麦克风并没有静音掉4、声卡驱动已重装过N次,新的旧的都试过了5、音量控制→属性→录音→麦克风下面的勾...

win10系统怎么分区(win10应该怎么分区)
  • win10系统怎么分区(win10应该怎么分区)
  • win10系统怎么分区(win10应该怎么分区)
  • win10系统怎么分区(win10应该怎么分区)
  • win10系统怎么分区(win10应该怎么分区)

取消回复欢迎 发表评论: