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

创作时间:

作者:

@小白创作中心

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

引用

CSDN

https://blog.csdn.net/ylong52/article/details/144204272

在处理图像时，去除图片中的文字是一个常见的需求，尤其是在处理扫描文档或照片时。本文将介绍如何使用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())

注：以上只有部分代码

步骤1：设置图像和矩形

首先，我们需要加载一张图片，并设置一个全局变量来存储所有选中的文字区域的矩形坐标。

步骤2：鼠标回调函数

我们定义了一个鼠标回调函数draw_rectangle，它允许用户通过鼠标点击和拖动来选择图片中的文字区域。这些区域将以矩形的形式被记录下来。

步骤3：删除和激活矩形

用户可以通过键盘快捷键来删除或激活某个矩形区域，以便对其进行编辑或调整。

步骤4：图像修复

在用户选择了所有文字区域后，我们使用OpenCV的inpaint函数来修复这些区域。这个函数会根据周围的图像内容来填充选中的区域，从而实现去除文字的效果。

步骤5：显示和保存结果

最后，我们将修复后的图像显示出来，并保存到磁盘上。