图像去字利器：教你用AI和OpenCV实现文字消除

黑金IT

已于 2024-12-03 09:36:40 修改

阅读量938

点赞数 15

分类专栏： python 文章标签：人工智能 opencv 计算机视觉 python

于 2024-12-03 09:31:17 首次发布

本文链接：https://blog.csdn.net/ylong52/article/details/144204272

版权

python 专栏收录该内容

28 篇文章

订阅专栏

如何使用Python和OpenCV去除图片中的文字

在处理图像时，去除图片中的文字是一个常见的需求，尤其是在处理扫描文档或照片时。本文将介绍如何使用Python编程语言和OpenCV库来实现这一功能。我们将通过一个简单的脚本，展示如何手动选择图片中的文字区域，并使用图像修复技术去除这些文字。

硬件要求

win11
无GPU要求
内存16G

环境准备

在开始之前，请确保你的环境中安装了Python和以下库：

OpenCV
NumPy
PIL（Python Imaging Library）
moviepy（可选，用于视频处理）

你可以通过pip安装这些库：

pip install opencv-python numpy pillow moviepy

代码解析

以下是我们的Python脚本，它包含了去除图片中文字的主要逻辑。

import asyncio
import cv2
import keyboard

# 全局变量
image_path = 'your_image_path.png'  # 替换为你的图片路径
original_image = None  # 原始图像
rects = []  # 存储所有矩形的坐标

# 鼠标回调函数
def draw_rectangle(event, x, y, flags, param):
    global ix, iy, drawing, original_image, rects
    # 省略部分代码...

# 删除激活的矩形并更新图像显示
def remove_active_rectangle():
    global rects, active_rect_index, original_image
    # 省略部分代码...

# 检测点是否在矩形的边界上
def is_point_on_rect_border(point, rect, thickness=3):
    # 省略部分代码...

# 主函数
async def main():
    await draw()
    global  original_image,active_rect_index,rects
    # 读取图片
    # image_path = 'cavity/c12.png'  # 替换为你的图片路径
    original_image = cv2.imread(image_path)
    cv2.namedWindow('image')
    # 设置鼠标回调函数
    cv2.setMouseCallback('image', draw_rectangle)

    # 主循环
    while True:
        cv2.imshow('image', original_image)
        key = cv2.waitKey(1) & 0xFF

        if key == 255 or key == 0:
            continue

        if keyboard.is_pressed('a') and keyboard.is_pressed('ctrl') and False:
            '''下面的代码不对，不能实现调整矩形的宽和高，暂时停止开发'''
            if active_rect_index >= 0:
                rect = rects[active_rect_index]
                # 缩小矩形，每次将高度减少1像素
                new_height = rect[1][1] - 1 if rect[1][1] - 1 > rect[0][1] else rect[0][1]
                new_rect = (rect[0], (rect[1][0], new_height))
                # 确保新的矩形坐标不超出原始图像边界
                new_rect = ((max(0, new_rect[0][0]), min(original_image.shape[0] - 1, new_rect[1][0])),
                            (max(0, new_rect[0][1]), min(original_image.shape[1] - 1, new_rect[1][1])))
                # 更新矩形列表中的激活矩形
                rects[active_rect_index] = new_rect
                # 重新绘制所有矩形到 original_image
                original_image_copy = original_image.copy()  # 创建 original_image 的副本
                for rect in rects:
                    cv2.rectangle(original_image_copy, rect[0], rect[1], (0, 255, 0), 3)
                # 更新 original_image 为最新的图像
                original_image = original_image_copy
                # 显示更新后的图像
                cv2.imshow('image', original_image)

        if keyboard.is_pressed('backspace') or keyboard.is_pressed('delete'):
            if active_rect_index >= 0:
                remove_active_rectangle()

        if key == 27:  # 按ESC键退出
            break


    # 销毁所有窗口
    cv2.destroyAllWindows()


if __name__ == '__main__':
    asyncio.run(main())