/keras-auto-augment

Keras implementation of AutoAugment.

Primary LanguagePythonMIT LicenseMIT

Keras implementation of AutoAugment

This repository contains code for AutoAugment (only using paper's best policies) based on AutoAugment: Learning Augmentation Policies from Data implemented in Keras.

example

Requirements

  • Python 3.6
  • Keras 2.2.4

Training

CIFAR-10

WideResNet28-10 baseline on CIFAR-10:

python train.py

WideResNet28-10 +Cutout, AutoAugment on CIFAR-10:

python train.py --cutout True --auto-augment True

Results

CIFAR-10

Model Accuracy (%) Loss
WideResNet28-10 baseline 95.32 0.3717
WideResNet28-10 +Cutout, AutoAugment 96.06 0.3565