No project description provided
Project description
pytorch-randaugment
Unofficial PyTorch Reimplementation of AutoAugment and RandAugment.
Code taken from https://github.com/DeepVoltaire/AutoAugment and https://github.com/jizongFox/uda
How to install:
pip install randaugment
How to use:
from autoaugment import RandAugment, ImageNetPolicy
data = ImageFolder(rootdir, transform=transforms.Compose(
[
transforms.RandomCrop(32, padding=4, fill=128), # fill parameter needs torchvision installed from source
transforms.RandomHorizontalFlip(),
RandAugment(),
#ImageNetPolicy(),
transforms.ToTensor(),
Cutout(size=16), # (https://github.com/uoguelph-mlrg/Cutout/blob/master/util/cutout.py)
transforms.Normalize(...)
])
)
loader = DataLoader(data, ...)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
randaugment-1.0.2.tar.gz
(5.4 kB
view hashes)
Built Distribution
Close
Hashes for randaugment-1.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d499cfdd4ea5443609df72eeac3a817146cc70f6544c6304ae1078dadaf99b98 |
|
MD5 | d4d0eb3403b22db4b0a85f35f145b7fe |
|
BLAKE2b-256 | fbeae24549f459800dc3bed21cd4e9c0d49d5b8deed65214b2444bd3e5a49f30 |