Train multiclass segmentation model on transformations

[suhacker1]

The current neural network for multiclass segmentation merely trains on the dataset obtained from the cloud. However, it would improve the performance of the system in the wild if it was also trained on transformations of the images in the original dataset. The albumentations library is a popular choice.

AC: multiclass segmentation model also trained on image transformations