Making presets public.

Author: datumbox

torchvision/prototype/models/alexnet.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.alexnet import AlexNet

torchvision/prototype/models/convnext.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, List, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.convnext import ConvNeXt, CNBlockConfig

torchvision/prototype/models/densenet.py

@@ -3,7 +3,7 @@
 from typing import Any, Optional, Tuple
 
 import torch.nn as nn
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.densenet import DenseNet

torchvision/prototype/models/detection/faster_rcnn.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional, Union
 
 from torch import nn
-from torchvision.prototype.transforms._presets import ObjectDetectionEval
+from torchvision.prototype.transforms import ObjectDetectionEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.detection.faster_rcnn import (

torchvision/prototype/models/detection/fcos.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional
 
 from torch import nn
-from torchvision.prototype.transforms._presets import ObjectDetectionEval
+from torchvision.prototype.transforms import ObjectDetectionEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.detection.fcos import (

torchvision/prototype/models/detection/keypoint_rcnn.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional
 
 from torch import nn
-from torchvision.prototype.transforms._presets import ObjectDetectionEval
+from torchvision.prototype.transforms import ObjectDetectionEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.detection.keypoint_rcnn import (

torchvision/prototype/models/detection/mask_rcnn.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional
 
 from torch import nn
-from torchvision.prototype.transforms._presets import ObjectDetectionEval
+from torchvision.prototype.transforms import ObjectDetectionEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.detection.mask_rcnn import (

torchvision/prototype/models/detection/retinanet.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional
 
 from torch import nn
-from torchvision.prototype.transforms._presets import ObjectDetectionEval
+from torchvision.prototype.transforms import ObjectDetectionEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.detection.retinanet import (

torchvision/prototype/models/detection/ssd.py

@@ -1,7 +1,7 @@
 import warnings
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ObjectDetectionEval
+from torchvision.prototype.transforms import ObjectDetectionEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.detection.ssd import (

torchvision/prototype/models/detection/ssdlite.py

@@ -3,7 +3,7 @@
 from typing import Any, Callable, Optional
 
 from torch import nn
-from torchvision.prototype.transforms._presets import ObjectDetectionEval
+from torchvision.prototype.transforms import ObjectDetectionEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.detection.ssdlite import (

torchvision/prototype/models/efficientnet.py

@@ -2,7 +2,7 @@
 from typing import Any, Optional, Sequence, Union
 
 from torch import nn
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.efficientnet import EfficientNet, MBConvConfig, FusedMBConvConfig, _efficientnet_conf

torchvision/prototype/models/googlenet.py

@@ -2,7 +2,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.googlenet import GoogLeNet, GoogLeNetOutputs, _GoogLeNetOutputs

torchvision/prototype/models/inception.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.inception import Inception3, InceptionOutputs, _InceptionOutputs

torchvision/prototype/models/mnasnet.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.mnasnet import MNASNet

torchvision/prototype/models/mobilenetv2.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.mobilenetv2 import MobileNetV2

torchvision/prototype/models/mobilenetv3.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional, List
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.mobilenetv3 import MobileNetV3, _mobilenet_v3_conf, InvertedResidualConfig

torchvision/prototype/models/optical_flow/raft.py

@@ -4,7 +4,7 @@
 from torch.nn.modules.instancenorm import InstanceNorm2d
 from torchvision.models.optical_flow import RAFT
 from torchvision.models.optical_flow.raft import _raft, BottleneckBlock, ResidualBlock
-from torchvision.prototype.transforms._presets import OpticalFlowEval
+from torchvision.prototype.transforms import OpticalFlowEval
 from torchvision.transforms.functional import InterpolationMode
 
 from .._api import WeightsEnum

torchvision/prototype/models/quantization/googlenet.py

@@ -2,7 +2,7 @@
 from functools import partial
 from typing import Any, Optional, Union
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.quantization.googlenet import (

torchvision/prototype/models/quantization/inception.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional, Union
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.quantization.inception import (

torchvision/prototype/models/quantization/mobilenetv2.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional, Union
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.quantization.mobilenetv2 import (

torchvision/prototype/models/quantization/mobilenetv3.py

@@ -2,7 +2,7 @@
 from typing import Any, List, Optional, Union
 
 import torch
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.quantization.mobilenetv3 import (

torchvision/prototype/models/quantization/resnet.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, List, Optional, Type, Union
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.quantization.resnet import (

torchvision/prototype/models/quantization/shufflenetv2.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, List, Optional, Union
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.quantization.shufflenetv2 import (

torchvision/prototype/models/regnet.py

@@ -2,7 +2,7 @@
 from typing import Any, Optional
 
 from torch import nn
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.regnet import RegNet, BlockParams

torchvision/prototype/models/resnet.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, List, Optional, Type, Union
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.resnet import BasicBlock, Bottleneck, ResNet

torchvision/prototype/models/segmentation/deeplabv3.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import SemanticSegmentationEval
+from torchvision.prototype.transforms import SemanticSegmentationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.segmentation.deeplabv3 import DeepLabV3, _deeplabv3_mobilenetv3, _deeplabv3_resnet

torchvision/prototype/models/segmentation/fcn.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import SemanticSegmentationEval
+from torchvision.prototype.transforms import SemanticSegmentationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.segmentation.fcn import FCN, _fcn_resnet

torchvision/prototype/models/segmentation/lraspp.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import SemanticSegmentationEval
+from torchvision.prototype.transforms import SemanticSegmentationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.segmentation.lraspp import LRASPP, _lraspp_mobilenetv3

torchvision/prototype/models/shufflenetv2.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.shufflenetv2 import ShuffleNetV2

torchvision/prototype/models/squeezenet.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.squeezenet import SqueezeNet

torchvision/prototype/models/vgg.py

@@ -1,7 +1,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.vgg import VGG, make_layers, cfgs

torchvision/prototype/models/video/resnet.py

@@ -2,7 +2,7 @@
 from typing import Any, Callable, List, Optional, Sequence, Type, Union
 
 from torch import nn
-from torchvision.prototype.transforms._presets import VideoClassificationEval
+from torchvision.prototype.transforms import VideoClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ....models.video.resnet import (

torchvision/prototype/models/vision_transformer.py

@@ -5,7 +5,7 @@
 from functools import partial
 from typing import Any, Optional
 
-from torchvision.prototype.transforms._presets import ImageClassificationEval
+from torchvision.prototype.transforms import ImageClassificationEval
 from torchvision.transforms.functional import InterpolationMode
 
 from ...models.vision_transformer import VisionTransformer, interpolate_embeddings  # noqa: F401

torchvision/prototype/transforms/__init__.py

@@ -10,4 +10,11 @@
 from ._geometry import HorizontalFlip, Resize, CenterCrop, RandomResizedCrop, FiveCrop, TenCrop, BatchMultiCrop
 from ._meta import ConvertBoundingBoxFormat, ConvertImageDtype, ConvertImageColorSpace
 from ._misc import Identity, Normalize, ToDtype, Lambda
+from ._presets import (
+    ObjectDetectionEval,
+    ImageClassificationEval,
+    SemanticSegmentationEval,
+    VideoClassificationEval,
+    OpticalFlowEval,
+)
 from ._type_conversion import DecodeImage, LabelToOneHot