- 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
lists anddicts. - The
input_typesargument must now be a PyTree that matches the structure of the inputs. - Default gamma range for the
RandomGammaaugmentation 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.