数据增强：随机擦除Random erasing

原创

已于 2022-03-29 15:58:29 修改 · 5.2k 阅读

24 ·

CC 4.0 BY-SA版权

文章标签：

#随机擦除 #数据增强 #random erasing

于 2022-03-25 17:08:51 首次发布

该方法的出处：《Random Erasing Data Augmentation》

论文链接：https://arxiv.org/pdf/1708.04896v2.pdf

代码链接：https://github.com/zhunzhong07/Random-Erasing

1.该方法操作后的效果图

在进行擦除过程中，在目标区域位置选择一个矩形区域，使用随机的像素值代替原始像素值，这样就对目标区域起到了一定的遮挡作用；

2.该方法的代码

from __future__ import absolute_import

from torchvision.transforms import *

import numpy as np
import torch

class RandomErasing(object):
    def __init__(self, EPSILON = 0.5, sl = 0.02, sh = 0.4, r1 = 0.3, mean=[0.4914, 0.4822, 0.4465]):
        self.EPSILON = EPSILON
        self.mean = mean
        self.sl = sl
        self.sh = sh
        self.r