[Bug Fix] Add missing transforms in __init__.py (#260)
parent
79d03b1597
commit
8d568fbc0d
|
@ -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'
|
||||
]
|
||||
|
|
Loading…
Reference in New Issue