Spaces:
Sleeping
Sleeping
Commit
·
964ef61
1
Parent(s):
74a0537
Fix typo
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ class SegmentationAlbumentationsTransform(ItemTransform):
|
|
26 |
def encodes(self, x):
|
27 |
img,mask = x
|
28 |
aug = self.aug(image=np.array(img), mask=np.array(mask))
|
29 |
-
return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
|
30 |
|
31 |
transforms=Compose([HorizontalFlip(p=0.5),
|
32 |
Rotate(p=0.40,limit=10),GridDistortion()
|
|
|
26 |
def encodes(self, x):
|
27 |
img,mask = x
|
28 |
aug = self.aug(image=np.array(img), mask=np.array(mask))
|
29 |
+
return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
|
30 |
|
31 |
transforms=Compose([HorizontalFlip(p=0.5),
|
32 |
Rotate(p=0.40,limit=10),GridDistortion()
|