Implementation of RandAugment in 3D
Project description
RandAugment-3D
Implementation of RandAugment in 3D
Original paper: arXiv:1909.13719
Original implementations (in tensorflow):
- From official paper: https://github.com/tensorflow/tpu/blob/298d1fa98638f302ab9df34d9d26bbded7220e8b/models/official/efficientnet/autoaugment.py
- More recent: https://github.com/tensorflow/models/blob/1c79ece9f43340e9bc9571e06a4bf9bd8db8d97a/official/vision/beta/ops/augment.py
Most functions used here are implemented by MONAI.
Installation
pip install RandAugment3d
How to use
Args:
- n: number of augmentations to apply
- magnitude: magnitude of augmentations to apply, on a scale where 10 is full level (this scale is used to be as similar as possible to the original implementation). Values > 10 result in more distortion, values < 10 in less.
- excluded_operations: list of names of the excluded operations. Valid names are listed above.
operations names: ['identity', 'rotate_x', 'rotate_y', 'rotate_z', 'translate', 'scale', 'shear_x', 'shear_y', 'shear_z', 'shiftIntensity', 'equalize', 'solarize', 'histogramShift', 'sharpen', 'adjustContrast']
augment = RandAugment3D(n=2, magnitude=10)
Geometric Augmentations only
augment = RandAugment3D(n=2, magnitude=10, excluded_operations=['shiftIntensity',
'equalize',
'solarize',
'histogramShift',
'sharpen',
'adjustContrast'])
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
Built Distribution
Close
Hashes for RandAugment3d-0.0.4.1.dev0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 15ff143515b3b906fd4d768c0020fb8cc99ef5864d46438500d846545e6a997e |
|
MD5 | 97b5113948126fd9da0a52e08c8660fa |
|
BLAKE2b-256 | 72c3f2e86dfb0ae10eee014d189521b9492b7eb21e16babeadf52d941007a63b |
Close
Hashes for RandAugment3d-0.0.4.1.dev0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9ee1164d5fde176a964efbe869a23e8f6c0e3007e22e1add92918653f5fa729b |
|
MD5 | d4a9d6154e9b2571cab794777f57c977 |
|
BLAKE2b-256 | ee156cff801b93de027eeeef9a12ce8bf6152a5aea4eb5375026aa57b40bc236 |