Skip to content

Commit 8d568fb

Browse files
authored
[Bug Fix] Add missing transforms in __init__.py (open-mmlab#260)
1 parent 79d03b1 commit 8d568fb

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

mmseg/datasets/pipelines/__init__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,13 @@
44
from .loading import LoadAnnotations, LoadImageFromFile
55
from .test_time_aug import MultiScaleFlipAug
66
from .transforms import (Normalize, Pad, PhotoMetricDistortion, RandomCrop,
7-
RandomFlip, Resize, SegRescale)
7+
RandomFlip, RandomRotate, Rerange, Resize, RGB2Gray,
8+
SegRescale)
89

910
__all__ = [
1011
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToDataContainer',
1112
'Transpose', 'Collect', 'LoadAnnotations', 'LoadImageFromFile',
1213
'MultiScaleFlipAug', 'Resize', 'RandomFlip', 'Pad', 'RandomCrop',
13-
'Normalize', 'SegRescale', 'PhotoMetricDistortion'
14+
'Normalize', 'SegRescale', 'PhotoMetricDistortion', 'RandomRotate',
15+
'Rerange', 'RGB2Gray'
1416
]

0 commit comments

Comments
 (0)