效果如图:
就是在图上加两个小黑块(1-2个)数量也可以自定义吧,然后也不一定是小黑块,颜色也可以自定义,具体看代码第6行
好像是没啥用哈,我是做目标检测数据增强用的
import numpy as np
from PIL import Image, ImageDraw
import random
def erase_random_areas(image, num_areas=1, min_size=90, max_size=150, background_color=(0, 0, 0)):
"""
擦除图像中的随机区域
:param image: 原始图像
:param num_areas: 擦除区域的数量
:param min_size: 擦除区域的最小尺寸
:param max_size: 擦除区域的最大尺寸
:param background_color: 背景颜色(擦除区域的填充颜色)
:return: 擦除后的图像
"""
image_copy = image.copy()
draw = ImageDraw.Draw(image_copy)
width, height = image_copy.size
for _ in range(n