MuarAugment 1.1.0

State-of-the-art data augmentation search algorithms in PyTorch

---

MuarAugment

Description

MuarAugment is a package providing the easiest way to a state-of-the-art data augmentation pipeline.

How to use

You can install MuarAugment via PIP:

!pip install muaraugment

Example (temp: tutorials and working examples coming soon)

from muar.augmentations import BatchRandAugment, MuAugment

# muar augmentations
rand_augment = BatchRandAugment(N_TFMS=3, MAGN=4)
mu_augment = MuAugment(rand_augment, N_COMPS=4, SELECTED=2)

# model
model = LitClassifier(mu_augment)

# data
train, val, test = mnist()

# train
trainer = Trainer()
trainer.fit(model, train, val)

Tutorials

• Overview of data augmentation policy search algorithms