← all repositories

kakaobrain/fast-autoaugment

PyTorch implementation of Fast AutoAugment, an AutoML technique for learning data augmentation policies to improve CNN training on image classification tasks.

1.6k stars Python ML FrameworksComputer Vision
fast-autoaugment
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

Fast AutoAugment is a NeurIPS 2019 research project that uses density-matching-based search to efficiently discover optimal data augmentation policies for training convolutional neural networks. The implementation learns augmentation strategies via a smarter search mechanism that is orders of magnitude faster than the original AutoAugment while achieving comparable performance across CIFAR and ImageNet benchmarks. It provides pre-trained model weights and supports distributed training.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.