PyRetri/pyretri/datasets/transformer/__init__.py

15 lines
472 B
Python

# -*- coding: utf-8 -*-
from yacs.config import CfgNode
from .transformers_impl.transformers import (DirectResize, PadResize, ShorterResize, CenterCrop,
ToTensor, ToCaffeTensor, Normalize, TenCrop, TwoFlip)
from .transformers_base import TransformerBase
__all__ = [
'TransformerBase',
'DirectResize', 'PadResize', 'ShorterResize', 'CenterCrop', 'ToTensor', 'ToCaffeTensor',
'Normalize', 'TenCrop', 'TwoFlip',
]