
摘要
本文介绍了一种新的卷积神经网络(CNN)训练数据增强方法——随机擦除(Random Erasing)。在训练过程中,随机擦除会随机选择图像中的一个矩形区域,并将其像素值替换为随机值。通过这一过程,可以生成具有不同程度遮挡的训练图像,从而降低过拟合的风险并提高模型对遮挡的鲁棒性。随机擦除无需参数学习,易于实现,并且可以与大多数基于CNN的识别模型集成。尽管方法简单,但随机擦除与常用的随机裁剪和翻转等数据增强技术相辅相成,并在图像分类、目标检测和行人重识别任务中对强大的基线模型表现出一致的性能提升。代码可从以下地址获取:https://github.com/zhunzhong07/Random-Erasing。
代码仓库
NVlabs/DG-Net
pytorch
pytorch/vision
pytorch
rwightman/pytorch-image-models
pytorch
GitHub 中提及
PaddlePaddle/PaddleClas
paddle
rlagywns0213/cifar100_data_augmentation
pytorch
GitHub 中提及
Janghyeonwoong/Random-Erasing-Tensorflow2
tf
GitHub 中提及
aditya30394/Person-Re-Identification
pytorch
GitHub 中提及
Brunogomes97/Imdb
GitHub 中提及
zhunzhong07/Random-Erasing
官方
pytorch
GitHub 中提及
rickyHong/Random-erasing-Data-Augumentation-repl
pytorch
GitHub 中提及
IyatomiLab/CE-CLCNN
pytorch
GitHub 中提及
CoinCheung/SphereReID
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| image-classification-on-fashion-mnist | Random Erasing | Percentage error: 3.65 |
| object-detection-on-pascal-voc-2007 | I+ORE | MAP: 76.2% |
| person-re-identification-on-dukemtmc-reid | TriNet + Random Erasing | Rank-1: 73.0 mAP: 56.6 |
| person-re-identification-on-dukemtmc-reid | SVDNet + Random Erasing | Rank-1: 79.3 mAP: 62.4 |
| robust-object-detection-on-cityscapes-1 | Cutout | mPC [AP]: 15.7 |