Ignaciobfp commited on
Commit
964ef61
·
1 Parent(s): 74a0537
Files changed (1) hide show
  1. app.py +1 -1
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()