Feature-level augmentation to improve robustness of deep neural networks to affine transformations