DEV Community

jaysonnguyen
jaysonnguyen

Posted on • Edited on

[BTY#1] 30/06/2023 Image Augmentation

How to Make Deep Models Strong and Robust with Data Augmentation

Refer to:

Problems of training Deep Learning models

  • Robustness

Image description

Image description

  • Generalization

Image description

Data Augmentation: Simple but very effective solution

  • Random Crop => handle scale and viewpoint change cases

Image description

  • Color jttering/lighting => handle color change cases

Image description

  • Random erasing/Cutout => handle occlusion cases

Image description

  • Mixup => handle uncertain images

Image description

  • CutMix => handle both occlusion and uncertain cases

Image description

Unlike Cutout, CutMix uses full image region
Unlike Mixup, CutMix makes realistic local image patches
CutMix is simple: only 20 lines of PyTorch code

Visualize CutMix effectiveness by heatmap

  • Mixup, Cutout and CutMix images

Image description

  • Heatmap results

Image description

Experiments with CutMix

  • For Image Classification

Image description

  • For Object Detection and Image Captioning

Image description

  • Improved robustness performance

Image description

Top comments (0)