mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
6a0a76af0c
* Add hparams argument in `AutoAugment` and `RandAugment`. And `pad_val` supports sequence instead of tuple only. * Add unit tests for `AutoAugment` and `hparams` in `RandAugment`. * Use smaller test image to speed up uni tests. * Use hparams to simplify RandAugment config in swin-transformer. * Rename augment config name from `pipeline` to `pipelines`. * Add some commnet ad docstring.
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
# Refers to `_RAND_INCREASING_TRANSFORMS` in pytorch-image-models
|
|
rand_increasing_policies = [
|
|
dict(type='AutoContrast'),
|
|
dict(type='Equalize'),
|
|
dict(type='Invert'),
|
|
dict(type='Rotate', magnitude_key='angle', magnitude_range=(0, 30)),
|
|
dict(type='Posterize', magnitude_key='bits', magnitude_range=(4, 0)),
|
|
dict(type='Solarize', magnitude_key='thr', magnitude_range=(256, 0)),
|
|
dict(
|
|
type='SolarizeAdd',
|
|
magnitude_key='magnitude',
|
|
magnitude_range=(0, 110)),
|
|
dict(
|
|
type='ColorTransform',
|
|
magnitude_key='magnitude',
|
|
magnitude_range=(0, 0.9)),
|
|
dict(type='Contrast', magnitude_key='magnitude', magnitude_range=(0, 0.9)),
|
|
dict(
|
|
type='Brightness', magnitude_key='magnitude',
|
|
magnitude_range=(0, 0.9)),
|
|
dict(
|
|
type='Sharpness', magnitude_key='magnitude', magnitude_range=(0, 0.9)),
|
|
dict(
|
|
type='Shear',
|
|
magnitude_key='magnitude',
|
|
magnitude_range=(0, 0.3),
|
|
direction='horizontal'),
|
|
dict(
|
|
type='Shear',
|
|
magnitude_key='magnitude',
|
|
magnitude_range=(0, 0.3),
|
|
direction='vertical'),
|
|
dict(
|
|
type='Translate',
|
|
magnitude_key='magnitude',
|
|
magnitude_range=(0, 0.45),
|
|
direction='horizontal'),
|
|
dict(
|
|
type='Translate',
|
|
magnitude_key='magnitude',
|
|
magnitude_range=(0, 0.45),
|
|
direction='vertical')
|
|
]
|