Use kornia library for data augmentation

[ziatdinovmax]

Currently, we have rather archaic data augmentation pipelines, which employ a combination of scikit-learn and open-cv image processing functions. This leads to a significant slow down of the training with on-the-fly data augmentation. It would be nice if we can instead use kornia computer vision library that was specifically designed for applications in deep learning.