__init__.py 520 Bytes
Newer Older
1
2
from torchvision import _BETA_TRANSFORMS_WARNING, _WARN_ABOUT_BETA_TRANSFORMS

3
from ._bounding_box import BoundingBoxes, BoundingBoxFormat
Philip Meier's avatar
Philip Meier committed
4
5
from ._datapoint import _FillType, _FillTypeJIT, _InputType, _InputTypeJIT
from ._image import _ImageType, _ImageTypeJIT, _TensorImageType, _TensorImageTypeJIT, Image
6
from ._mask import Mask
Philip Meier's avatar
Philip Meier committed
7
from ._video import _TensorVideoType, _TensorVideoTypeJIT, _VideoType, _VideoTypeJIT, Video
8

9
10
11
12
if _WARN_ABOUT_BETA_TRANSFORMS:
    import warnings

    warnings.warn(_BETA_TRANSFORMS_WARNING)