← all repositories

ildoonet/pytorch-randaugment

A PyTorch implementation of RandAugment data augmentation for training computer vision models on CIFAR, ImageNet, and SVHN.

637 stars Python Computer VisionML Frameworks
pytorch-randaugment
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository provides an unofficial PyTorch reimplementation of RandAugment, a data augmentation technique originally developed by Google for training deep learning models. It enables inserting augmentation transforms into standard image classification pipelines by specifying two hyperparameters (N and M). The library supports training on CIFAR-10/100, SVHN, and ImageNet datasets, and reproduces the competitive performance of the original RandAugment paper when applied to Wide-ResNet and other CNN architectures.

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