- Fixes:
- Fixed: Fixed RandomBrightness color adjustment for cases when the augmentation was not meant to be applied
- Fixes:
- Fixed: No longer running jax code on package import. Thanks @kvablack!
- Fixes:
- Fixed: Passing differently sized inputs to geometric transforms used to produce weird behaviour. Now this is correctly detected and a more helpful error message is thrown.
- Fixed: Typing hints updated to be compatible with jax >=0.4.24
- Bugfixes:
- Fixed: Inverting a transformation without given input types
- Fixed: RandomChannelGamma now correctly applies different gamma to each channel.
- Changes:
- Arguments to the augment function can now be arbitrary PyTrees, this includes
list
s anddict
s. - The
input_types
argument must now be a PyTree that matches the structure of the inputs. - Default gamma range for the
RandomGamma
augmentation was changed from[0.75, 1.33]
to[0.25, 4.0]
- Arguments to the augment function can now be arbitrary PyTrees, this includes
- New Augmentations:
- ChannelDrop: Drop a random channel from the image.
- RandomChannelGamma: Apply a separate Gamma transform to each channel of the image.