torch data augmentation toolbox (supports affine transform) - View it on GitHub