Update app.py

app.py

@@ -16,6 +16,12 @@ from albumentations import (
     ShiftScaleRotate
 )
 
+def get_y_fn (x):
+    return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))
+
+def ParentSplitter(x):
+    return Path(x).parent.name==test_name
+
 class SegmentationAlbumentationsTransform(ItemTransform):
     split_idx = 0
     
@@ -27,6 +33,11 @@ class SegmentationAlbumentationsTransform(ItemTransform):
         aug = self.aug(image=np.array(img), mask=np.array(mask))
         return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
 
+transforms=Compose([HorizontalFlip(p=0.5),
+                    Rotate(p=0.40,limit=10),GridDistortion()
+                    ],p=1)
+transformPipeline=SegmentationAlbumentationsTransform(transforms)
+
 class TargetMaskConvertTransform(ItemTransform):
     def __init__(self): 
         pass