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

利用CnOCR实现自动对截屏图片OCR

off999 2024-12-03 17:15 14 浏览 0 评论

cnocr的介绍可见:

更好用的开源Python OCR工具包:CnOCR V2.2


自动对截屏图片进行文字识别(OCR)

我自己有对截屏图片进行文字识别的需求,比如遇到某些网站不让拷贝,或者识别分享视频中的某个文字信息等。之前我都是用QQ自带的截屏识别功能来完成整个操作的,但是流程比较麻烦。


为什么不装一个对应的app?大部分免费的app后面其实都是调用的大公司的AI平台的接口,还得去折腾平台app key,怪麻烦的而且还无法离线使用。要么就是限制调用次数。当然,另一个原因是我一直期望用自己开发的 cnocr 来完成这个功能,eat your own dog food 嘛,自己不用的东西肯定没前途。


之前考虑过搞个浏览器插件或者做个系统应用来完成这个功能,但我对系统应用开发一窍不通而且前端技术也只了解基础的html和css,所以还是放弃了。欢迎大家朝着这个方向做些工作。


现在的实现方法很简单,各个系统(Windows、Mac、Linux)的截屏应用有很多很好用的,基本都有把截屏图片存储在指定位置这个功能。基于这个功能,我只要搞个守护进程轮询去查看是否有刚更新的图片,如果有就把它拿给cnocr识别下,然后把识别的结果复制进系统的剪切板即可。


怎么搭起来这个功能呢?开始之前先确保你的 cnocr v2.2版已经安装成功,否则请参考 安装文档 搞定cnocr最新版的安装。


以下是具体的搭建流程:

1. 找一个用得惯的截屏软件,这个软件只要支持把截屏图片存储在指定位置即可。比如Mac下我用的是免费的 Xnip

2. 除了安装cnocr v2.2外,这里面需要额外安装一个python包 pyperclip,利用它把ocr结果复制进系统的剪切板:

> pip install pyperclip

3. 拷贝下面的代码,存储到新文件 screenshot_daemon.py 。编辑此文件 "SCREENSHOT_DIR" 所在行,把路径改为你的截屏图片所存储的目录。

# coding: utf-8
# Copyright (C) 2022, [Breezedeus](https://github.com/breezedeus).

import os
import time
import glob
from pprint import pformat

import pyperclip as pc
from cnocr import CnOcr
from cnocr.utils import set_logger

logger = set_logger(log_level='DEBUG')


SCREENSHOT_DIR = os.getenv(
    "SCREENSHOT_DIR", '<你的截屏图片存储目录>'  # ==> 换成你的目录
)
OCR_MODEL = CnOcr()


def get_newest_fp_time(screenshot_dir):
    fn_list = glob.glob1(screenshot_dir, '*g')
    fp_list = [os.path.join(screenshot_dir, fn) for fn in fn_list]
    if not fp_list:
        return None, None
    fp_list.sort(key=lambda fp: os.path.getmtime(fp), reverse=True)
    return fp_list[0], os.path.getmtime(fp_list[0])


def ocr_newest(screenshot_dir, delta_interval):
    while True:
        newest_fp, newest_mod_time = get_newest_fp_time(screenshot_dir)
        if (
            newest_mod_time is not None
            and time.time() - newest_mod_time < delta_interval
        ):
            logger.info(f'analyzing screenshot file {newest_fp}')
            result = OCR_MODEL.ocr(newest_fp)
            texts = [_one['text'] for _one in result]
            logger.info(f'\tOCR results: {pformat(texts)}\n\n')
            if texts:
                pc.copy('\n'.join(texts))
        time.sleep(1)


if __name__ == '__main__':
    ocr_newest(SCREENSHOT_DIR, 2)

4. 运行此脚本文件,all done:

> python screenshot_daemon.py


接下来就是享受随时ocr的乐趣了:


欢迎大家尝试,遇到问题可以评论,感谢。


知识星球私享群

此文初始发表于 知识星球CnOCR/CnSTD私享群 (
https://wx.zsxq.com/dweb2/index/group/28858522821151)
知识星球私享群会陆续发布一些CnOCR/CnSTD相关的私有资料,包括更详细的训练教程各种资源福利,使用过程中遇到的疑难解答等。本群也会发布OCR/STD相关的最新研究资料。欢迎感兴趣的朋友加入。

相关推荐

独家 | 5 个Python高级特性让你在不知不觉中成为Python高手

你已经使用Python编程了一段时间,编写脚本并解决各种问题。是你的水平出色吗?你可能只是在不知不觉中利用了Python的高级特性。从闭包(closure)到上下文管理器(contextmana...

Python装饰器

Python装饰器是一种用于修改函数或类的行为的特殊语法。它们允许在不修改原始代码的情况下,通过将函数或类作为参数传递给另一个函数来添加额外的功能。装饰器本质上是一个函数,它接受一个函数作为参数,并返...

中高阶Python常规用法--上下文管理器

Python以简单性和通用性著称,是一种深受全球开发人员喜爱的编程语言。它提供了大量的特性和功能,使编码成为一种愉快的体验。在这些功能中,一个经常被新手忽视的强大工具是上下文管理器。上下文管理器是高...

Python小案例67- 装饰器

Python装饰器是一种用于修改函数或类的行为的特殊语法。它们允许在不修改原始代码的情况下,通过将函数或类作为参数传递给另一个函数来添加额外的功能。装饰器本质上是一个函数,它接受一个函数作为参数,并返...

python常用的语法糖

概念Python的语法糖(SyntacticSugar)是指那些让代码更简洁、更易读的语法特性,它们本质上并不会增加新功能,但能让开发者更高效地编写代码。推导式写法推导式是Python最经典的...

python - 常用的装饰器 decorator 有哪些?

python编程中使用装饰器(decorator)工具,可以使代码更简洁清晰,提高代码的重用性,还可以为代码维护提供方便。对于python初学者来说,根据装饰器(decorator)的字面意思并不...

python数据缓存怎么搞 ?推荐一个三方包供你参考,非常简单好用。

1.数据缓存说明数据缓存可以说也是项目开发中比不可少的一个工具,像我们测试的系统中,你都会见到像Redis一样的数据缓存库。使用缓存数据库的好处不言而喻,那就是效率高,简单数据直接放在缓存中...

用于时间序列数据的Graphite监视工具

结合第三方工具,Graphite为IT性能监控提供了许多好处。本文介绍其核心组件,包括Carbon、Whisper以及安装的基本准则。Graphite监视工具可实时或按需,大规模地绘制来自多个来源的时...

Python3+pygame实现的坦克大战

一、显示效果二、代码1.说明几乎所有pygame游戏,基本都遵循一定的开发流程,大体如下:初始化pygame创建窗口while循环检测以及处理事件(鼠标点击、按键等)更新UI界面2.代码创建一个m...

Python之鸭子类型:一次搞懂with与上下文装饰器

引言在鸭子类型的理念的基础之上,从关注类型,转变到关注特性和行为。结合Python中的魔法函数的体系,我们可以将自定义的类型,像内置类型一样被使用。今天这篇文章中,接着该话题,继续聊一下with语法块...

Python必会的50个代码操作

学习Python时,掌握一些常用的程序操作非常重要。以下是50个Python必会的程序操作,主要包括基础语法、数据结构、函数和文件操作等。1.HelloWorldprint("Hello,...

一文掌握Python 中的同步和异步

同步代码(Sync)同步就像在一个流水线上工作,每个任务都等待前一个任务完成。示例:机器A切割钢板→完成后,机器B钻孔→完成后,机器C上色。在Python中,同步代码看起来像这样:im...

python 标注模块timeit: 测试函数的运行时间

在Python中,可以使用内置的timeit模块来测试函数的运行时间。timeit模块提供了一个简单的接口来测量小段代码的执行时间。以下是使用timeit测试函数运行时间的一般步骤:导入...

Python带你找回童年的万花尺

还记得小时候的万花尺吧?这么画:一点也不费脑筋,就可以出来这么多丰富多彩的复杂几何图形。具体而言,可以用万花尺玩具(如图2-1所示)来绘制数学曲线。这种玩具由两个不同尺寸的塑料齿轮组成,一大一小。小的...

Python 时间模块深度解析:从基础到高级的全面指南

直接上干货一、时间模块核心类介绍序号类名说明1datetime.datetime表示一个具体的日期和时间,结合了日期和时间的信息。2datetime.date表示一个具体的日期。3datetime.t...

取消回复欢迎 发表评论: