mmpretrain/mmcls/datasets/pipelines/__init__.py

14 lines
576 B
Python

from .compose import Compose
from .formating import (Collect, ImageToTensor, ToDataContainer, ToNumpy,
ToPIL, ToTensor, Transpose, to_tensor)
from .loading import LoadImageFromFile
from .transforms import (CenterCrop, Normalize, RandomHorizontalFlip,
RandomResizedCrop, Resize)
__all__ = [
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToDataContainer',
'ToPIL', 'ToNumpy', 'Transpose', 'Collect', 'LoadImageFromFile',
'RandomResizedCrop', 'RandomHorizontalFlip', 'Resize', 'CenterCrop',
'Normalize'
]