Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,17 @@ from fastai.vision.all import *
|
|
4 |
import PIL
|
5 |
import torchvision.transforms as transforms
|
6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
class TargetMaskConvertTransform(ItemTransform):
|
8 |
def __init__(self):
|
9 |
pass
|
|
|
4 |
import PIL
|
5 |
import torchvision.transforms as transforms
|
6 |
|
7 |
+
class SegmentationAlbumentationsTransform(ItemTransform):
|
8 |
+
split_idx = 0
|
9 |
+
|
10 |
+
def __init__(self, aug):
|
11 |
+
self.aug = aug
|
12 |
+
|
13 |
+
def encodes(self, x):
|
14 |
+
img,mask = x
|
15 |
+
aug = self.aug(image=np.array(img), mask=np.array(mask))
|
16 |
+
return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
|
17 |
+
|
18 |
class TargetMaskConvertTransform(ItemTransform):
|
19 |
def __init__(self):
|
20 |
pass
|