Skip to content

Commit a24aae4

Browse files
committed
Removes vertical flip: horizontal plus n*90 degree equivalent
1 parent 8b7566e commit a24aae4

1 file changed

Lines changed: 0 additions & 2 deletions

File tree

robosat/tools/train.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
from robosat.transforms import (
1818
JointCompose,
1919
JointTransform,
20-
JointRandomVerticalFlip,
2120
JointRandomHorizontalFlip,
2221
JointRandomRotation,
2322
ConvertImageMode,
@@ -185,7 +184,6 @@ def get_dataset_loaders(model, dataset):
185184
JointTransform(ConvertImageMode("RGB"), ConvertImageMode("P")),
186185
JointTransform(Resize(target_size, Image.BILINEAR), Resize(target_size, Image.NEAREST)),
187186
JointTransform(CenterCrop(target_size), CenterCrop(target_size)),
188-
JointRandomVerticalFlip(0.5),
189187
JointRandomHorizontalFlip(0.5),
190188
JointRandomRotation(0.5, 90),
191189
JointRandomRotation(0.5, 180),

0 commit comments

Comments
 (0)