- Changes: Relaxed
numpyversion requirement
- Fixes:
- Change jax.tree_map to jax.tree_util.tree_map. (#11 by @mdenil)
- Changes:
- Now using
uvfor project management and building the package
- Now using
- 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
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.