Skip to content

Commit 5ac4e6c

Browse files
authored
remove usage of deprecated albumentations Flip() transform (#2318)
1 parent f8866af commit 5ac4e6c

File tree

1 file changed

+4
-2
lines changed
  • rastervision_pytorch_backend/rastervision/pytorch_backend/examples/semantic_segmentation

1 file changed

+4
-2
lines changed

rastervision_pytorch_backend/rastervision/pytorch_backend/examples/semantic_segmentation/utils.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@
22
import albumentations as A
33

44
example_rgb_transform = A.Compose([
5-
A.Flip(),
5+
A.VerticalFlip(),
6+
A.HorizontalFlip(),
67
A.Transpose(),
78
A.RandomRotate90(),
89
A.ShiftScaleRotate(),
@@ -38,7 +39,8 @@
3839
# not all transforms work with more than 3 channels, here are
3940
# some of the ones that do
4041
example_multiband_transform = A.Compose([
41-
A.Flip(),
42+
A.VerticalFlip(),
43+
A.HorizontalFlip(),
4244
A.Transpose(),
4345
A.RandomRotate90(),
4446
A.ShiftScaleRotate(),

0 commit comments

Comments
 (0)