OpenCV实现对图片ROI的随机擦除（个人笔记）

关注

VIP免费原创最新推荐文章于 2026-10-14 14:59:38 发布 1.5k 阅读 AI 写同款· GEO 优化 ›

python opencv 已知图片和mask，抠出mask区域目录前言：目标：思路：完整代码：前言在ICDAR2021竞赛中，需要取出图片中需要分割的样本，对样本进行仿射变换，背景替换，生成更丰富的样本。目标结合mask取出input中的两张卡片 input mask ... 阅读详情

import numpy as np
import cv2 as cv
import random


def mask_opration(src_image, location, mask_image):
    """
    把mask_iamge根据mask_shape添加到目标图片上
    :param src_image:
    :param location:需要添加到的位置框，(xmin, ymin, xmax, ymax)
    :param mask_image: 彩色图
    :return:
    """
    # 根据mask_shape获取roi
    left, top, right, bottom = location
    roi = src_image[top:bottom, left:right]

    grid_num = np.random.randint(3, 5)
    mask_shape = grid_mask(location, grid_num)
    _, center_mask = cv.threshold(mask_shape, 125, 255, cv.THRESH_BINARY)  # 为了确保是二值化图片
    center_mask = np.tile(np.expand_dims(center_mask, axis=-1), [1, 1, 3])  # 位操作间维度保持一致
    mask_image = cv.bitwise_and(mask_image, center_mask)

    edge_mask = cv.bitwise_not(center_mask)
    mask_inv_image = cv.bitwise_and(roi, edge_mask)
    mask = cv.bitwise_xor(mask_image, mask_inv_image)
    src_image[top:bottom, left:right] = mask
    return src_image


def grid_mask(roi, grid_num):
    xmin,