mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

[Bug Fix] Add missing transforms in __init__.py (#260)

Browse Source
This commit is contained in:
Jerry Jiarui XU 2020-11-17 20:22:46 -08:00 committed by GitHub
parent a6234975f1
commit 97f10dbb04
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mmseg/datasets/pipelines/__init__.py
View File

@ -4,11 +4,13 @@ from .formating import (Collect, ImageToTensor, ToDataContainer, ToTensor,
from .loading import LoadAnnotations, LoadImageFromFile
from .test_time_aug import MultiScaleFlipAug
from .transforms import (Normalize, Pad, PhotoMetricDistortion, RandomCrop,
RandomFlip, Resize, SegRescale)
RandomFlip, RandomRotate, Rerange, Resize, RGB2Gray,
SegRescale)
__all__ = [
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToDataContainer',
'Transpose', 'Collect', 'LoadAnnotations', 'LoadImageFromFile',
'MultiScaleFlipAug', 'Resize', 'RandomFlip', 'Pad', 'RandomCrop',
'Normalize', 'SegRescale', 'PhotoMetricDistortion'
'Normalize', 'SegRescale', 'PhotoMetricDistortion', 'RandomRotate',
'Rerange', 'RGB2Gray'
]