expose api

dd0aa320 · LielinJiang · 78ecceb2 · dd0aa320
隐藏空白更改
内联并排

Showing with 18 addition and 0 deletion

transform/transforms.py transform/transforms.py +18 -0

未找到文件。
--- a/transform/transforms.py
+++ b/transform/transforms.py
@@ -20,6 +20,24 @@ else:
    Sequence = collections.abc.Sequence
    Iterable = collections.abc.Iterable

+__all__ = [
+    "Compose",
+    "Resize",
+    "RandomResizedCrop",
+    "CenterCropResize",
+    "CenterCrop",
+    "RandomHorizontalFlip",
+    "RandomVerticalFlip",
+    "Permute",
+    "Normalize",
+    "GaussianNoise",
+    "BrightnessTransform",
+    "SaturationTransform",
+    "ContrastTransform",
+    "HueTransform",
+    "ColorJitter",
+]
+

 class Compose(object):
    """Composes several transforms together.