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

最新推荐文章于 2023-08-10 22:08:53 发布

vector_Lu

最新推荐文章于 2023-08-10 22:08:53 发布

阅读量1.3k

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/vector_lu/article/details/108128507

版权

本文详细介绍了如何利用OpenCV库在图像处理中实现对感兴趣区域（ROI）的随机擦除，这一技术常用于数据增强，提高模型的泛化能力。通过实例代码，展示了具体的实现步骤和关键函数的应用。

摘要由CSDN通过智能技术生成

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:r