python opencv 按点抠图

原创

mob64ca12df9869 2024-09-06 05:35:05 ©著作权

文章标签 抠图 Image OpenCV 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12df9869的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用 Python 和 OpenCV 实现按点抠图

在图像处理领域，抠图是一个基本且重要的任务。使用 Python 和 OpenCV 库可以方便地实现按点抠图。本文将引导你一步一步地实现这一功能，并提供必要的代码和详细解释。我们还将用表格展示整个流程，并使用 mermaid 语法显示关系图和类图。

流程概述

在进行按点抠图前，我们需要明确整个过程的步骤。以下是相关流程的表格：

步骤	详细描述
1. 导入库	导入所需的 Python 库
2. 读取图像	使用 OpenCV 读取图像
3. 定义掩膜	创建一个与图像相同大小的掩膜，初始化为全黑
4. 选择点	用户逐点标记需要抠取的部分
5. 应用掩膜	使用掩膜对图像进行抠图
6. 显示结果	展示结果图像和保存图像

步骤详细说明

1. 导入库

import cv2     # 导入 OpenCV 库
import numpy as np  # 导入 NumPy 库

上面的代码分别导入了 OpenCV 用于图像处理和 NumPy 用于处理数组的库。

2. 读取图像

image = cv2.imread('image.jpg')  # 读取图像文件

cv2.imread 函数用于读取指定路径的图像，并将其存储在变量 image 中。

3. 定义掩膜

mask = np.zeros(image.shape[:2], dtype=np.uint8)  # 创建与图像大小相同的全黑掩膜

使用 NumPy 创建一个与输入图像相同大小的全零数组（黑色掩膜），以便后续用于扣取图像。

4. 选择点

用户需要通过鼠标点击来选择需要抠图的点。我们可以借助 OpenCV 的回调函数来实现。

points = []  # 存储鼠标点击的点

def select_point(event, x, y, flags, param):
    if event == cv2.EVENT_LBUTTONDOWN:  # 如果按下了鼠标左键
        points.append((x, y))  # 将点击的点加入列表
        cv2.circle(image, (x, y), 5, (0, 255, 0), -1)  # 在图像中画出点击的点
        cv2.imshow('Image', image)  # 显示图像

cv2.namedWindow('Image')  # 创建窗口
cv2.setMouseCallback('Image', select_point)  # 设置鼠标回调函数
cv2.imshow('Image', image)  # 显示图像
cv2.waitKey(0)  # 等待按键

上述代码定义了 select_point 函数，用于捕获用户的鼠标点击事件并在图像上绘制出绿色圆点。

5. 应用掩膜

for point in points:
    cv2.circle(mask, point, radius=5, color=255, thickness=-1)  # 在掩膜上绘制白色圆圈

result = cv2.bitwise_and(image, image, mask=mask)  # 使用掩膜抠图

这段代码迭代用户选择的所有点，并在掩膜上绘制出白色圆圈。然后使用 cv2.bitwise_and 函数将掩膜应用于原图，从而实现抠图。

6. 显示结果

cv2.imshow('Result', result)  # 显示抠图结果
cv2.imwrite('result.jpg', result)  # 将结果保存到文件
cv2.waitKey(0)  # 等待按键
cv2.destroyAllWindows()  # 关闭所有窗口

最后一段代码将抠图结果进行展示，并保存为 result.jpg 文件。

关系图

下面是使用 mermaid 语法表示的图像处理过程中的关系图：

erDiagram
    Image {
        string path
        string format
        int width
        int height
    }
    Mask {
        int width
        int height
    }
    User {
        string action
    }
    User ||--o{ Image : interacts_with
    User }o--o{ Mask : creates
    Image ||--o{ Mask : applies_to

类图

下面是使用 mermaid 语法表示的简单类图：

classDiagram
    class Image {
        +load(path)
        +display()
    }
    class Mask {
        +create(size)
        +apply(image)
    }
    class User {
        +selectPoints()
        +saveImage()
    }
    User --> Image : interacts
    User --> Mask : creates