Update app.py
Browse files
app.py
CHANGED
@@ -12,18 +12,7 @@ from pathlib import Path
|
|
12 |
import random
|
13 |
import PIL
|
14 |
import torchvision.transforms as transforms
|
15 |
-
from albumentations import
|
16 |
-
Compose,
|
17 |
-
OneOf,
|
18 |
-
ElasticTransform,
|
19 |
-
GridDistortion,
|
20 |
-
OpticalDistortion,
|
21 |
-
HorizontalFlip,
|
22 |
-
Rotate,
|
23 |
-
Transpose,
|
24 |
-
CLAHE,
|
25 |
-
ShiftScaleRotate
|
26 |
-
)
|
27 |
|
28 |
def get_y_fn (x):
|
29 |
return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))
|
|
|
12 |
import random
|
13 |
import PIL
|
14 |
import torchvision.transforms as transforms
|
15 |
+
from albumentations import *
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
|
17 |
def get_y_fn (x):
|
18 |
return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))
|