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

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

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

相关推荐

让 Python 代码飙升330倍:从入门到精通的四种性能优化实践

花下猫语:性能优化是每个程序员的必修课,但你是否想过,除了更换算法,还有哪些“大招”?这篇文章堪称典范,它将一个普通的函数,通过四套组合拳,硬生生把性能提升了330倍!作者不仅展示了“术”,更传授...

7 段不到 50 行的 Python 脚本,解决 7 个真实麻烦:代码、场景与可复制

“本文整理自开发者AbdurRahman在Stackademic的真实记录,所有代码均经过最小化删减,确保在50行内即可运行。每段脚本都对应一个日常场景,拿来即用,无需额外依赖。一、在朋...

Python3.14:终于摆脱了GIL的限制

前言Python中最遭人诟病的设计之一就是GIL。GIL(全局解释器锁)是CPython的一个互斥锁,确保任何时刻只有一个线程可以执行Python字节码,这样可以避免多个线程同时操作内部数据结...

Python Web开发实战:3小时从零搭建个人博客

一、为什么选Python做Web开发?Python在Web领域的优势很突出:o开发快:Django、Flask这些框架把常用功能都封装好了,不用重复写代码,能快速把想法变成能用的产品o需求多:行业...

图解Python编程:从入门到精通系列教程(附全套速查表)

引言本系列教程展开讲解Python编程语言,Python是一门开源免费、通用型的脚本编程语言,它上手简单,功能强大,它也是互联网最热门的编程语言之一。Python生态丰富,库(模块)极其丰富,这使...

Python 并发编程实战:从基础到实战应用

并发编程是提升Python程序效率的关键技能,尤其在处理多任务场景时作用显著。本文将系统介绍Python中主流的并发实现方式,帮助你根据场景选择最优方案。一、多线程编程(threading)核...

吴恩达亲自授课,适合初学者的Python编程课程上线

吴恩达教授开新课了,还是亲自授课!今天,人工智能著名学者、斯坦福大学教授吴恩达在社交平台X上发帖介绍了一门新课程——AIPythonforBeginners,旨在从头开始讲授Python...

Python GUI 编程:tkinter 初学者入门指南——Ttk 小部件

在本文中,将介绍Tkinter.ttk主题小部件,是常规Tkinter小部件的升级版本。Tkinter有两种小部件:经典小部件、主题小部件。Tkinter于1991年推出了经典小部件,...

Python turtle模块编程实践教程

一、模块概述与核心概念1.1turtle模块简介定义:turtle是Python标准库中的2D绘图模块,基于Logo语言的海龟绘图理念实现。核心原理:坐标系系统:原点(0,0)位于画布中心X轴:向右...

Python 中的asyncio 编程入门示例-1

Python的asyncio库是用于编写并发代码的,它使用async/await语法。它为编写异步程序提供了基础,通过非阻塞调用高效处理I/O密集型操作,适用于涉及网络连接、文件I/O...

30天学会Python,开启编程新世界

在当今这个数字化无处不在的时代,Python凭借其精炼的语法架构、卓越的性能以及多元化的应用领域,稳坐编程语言排行榜的前列。无论是投身于数据分析、人工智能的探索,还是Web开发的构建,亦或是自动化办公...

Python基础知识(IO编程)

1.文件读写读写文件是Python语言最常见的IO操作。通过数据盘读写文件的功能都是由操作系统提供的,读写文件就是请求操作系统打开一个文件对象(通常称为文件描述符),然后,通过操作系统提供的接口从这个...

Python零基础到精通,这8个入门技巧让你少走弯路,7天速通编程!

Python学习就像玩积木,从最基础的块开始,一步步搭建出复杂的作品。我记得刚开始学Python时也是一头雾水,走了不少弯路。现在回头看,其实掌握几个核心概念,就能快速入门这门编程语言。来聊聊怎么用最...

一文带你了解Python Socket 编程

大家好,我是皮皮。前言Socket又称为套接字,它是所有网络通信的基础。网络通信其实就是进程间的通信,Socket主要是使用IP地址,协议,端口号来标识一个进程。端口号的范围为0~65535(用户端口...

Python-面向对象编程入门

面向对象编程是一种非常流行的编程范式(programmingparadigm),所谓编程范式就是程序设计的方法论,简单的说就是程序员对程序的认知和理解以及他们编写代码的方式。类和对象面向对象编程:把...

取消回复欢迎 发表评论: