--- a
+++ b/data/augmentation.py
@@ -0,0 +1,42 @@
+import albumentations as A
+from torchvision import transforms
+
+
+def pipeline_tranforms():
+    return transforms.Compose([
+        transforms.Resize((512, 512)),
+        transforms.ToTensor()
+    ])
+
+
+def albumentation_pixelwise_transforms():
+    return A.Compose([
+        A.ImageCompression(1, 50, p=0.1),
+        A.ColorJitter(brightness=0.2, contrast=0.2, saturation=0.2, hue=0.01, p=0.1),
+        A.RandomSnow(p=0.1),
+        A.GlassBlur(p=0.1),
+        A.CLAHE(p=0.1)
+
+    ])
+
+
+def albumentation_mask_transforms():
+    return A.Compose([
+        A.HorizontalFlip(),
+        A.VerticalFlip(),
+        A.RandomRotate90(p=0.25),
+        A.ElasticTransform(p=0.1),
+        A.OpticalDistortion(p=0.1)
+    ])
+
+
+def image_transforms():
+    return transforms.Compose([
+        transforms.ColorJitter(0.1, 0.1, 0.1),
+        transforms.RandomAdjustSharpness(0.9, 0.1),
+        transforms.RandomAutocontrast()
+    ])
+
+
+if __name__ == '__main__':
+    pass