solt 0.2.1__py3-none-any.whl

solt/__init__.py

@@ -0,0 +1,21 @@
+__version__ = "0.2"
+
+from . import constants, transforms, core, utils
+
+from .core import Stream, SelectiveStream
+from .core import DataContainer, Keypoints
+from .utils import from_yaml, from_dict, from_json
+
+__all__ = [
+    "constants",
+    "transforms",
+    "core",
+    "utils",
+    "from_dict",
+    "from_json",
+    "from_yaml",
+    "Stream",
+    "SelectiveStream",
+    "DataContainer",
+    "Keypoints",
+]

solt/constants/__init__.py

@@ -0,0 +1,56 @@
+"""
+Constants module
+
+.. data:: ALLOWED_BLURS
+
+    Defines the blur modes. Can be ``'g'`` - gaussian or ``'m'`` - median.
+
+.. data:: ALLOWED_TYPES
+
+    Defines the allowed types to be stored in a DataCantainer. Can be
+    ``'I'`` - image, ``'M'`` - mask, ``'L'`` - labels, ``'P'`` - Keypoints.
+
+.. data:: ALLOWED_CROPS
+
+    Defines the allowed crops. Can be ``'r'`` - random crop or ``'c'`` - center crop.
+
+.. data:: ALLOWED_PADDINGS
+
+    Defines the allowed crops. Can be ``'z'`` - zero padding or ``'r'`` - reflective padding.
+
+.. data:: ALLOWED_INTERPOLATIONS
+
+    Defines the allowed interpolation modes. Can be ``'bilinear'``, ``'nearest'`` or ``'bicubic'``.
+
+.. data:: DTYPES_MAX
+
+    Defines the maximums for different data types. Can be ``numpy.uint8`` or ``numpy.uint16``.
+
+.. data:: ALLOWED_COLOR_CONVERSIONS
+
+    Defines the allowed color conversion modes. Can be ``'gs2rgb'``, ``'rgb2gs'`` or ``'none'``.
+
+
+"""
+
+from ._constants import (
+    ALLOWED_BLURS,
+    ALLOWED_TYPES,
+    ALLOWED_CROPS,
+    ALLOWED_PADDINGS,
+    ALLOWED_INTERPOLATIONS,
+    DTYPES_MAX,
+    ALLOWED_COLOR_CONVERSIONS,
+    ALLOWED_GRIDMASK_MODES,
+)
+
+__all__ = [
+    "ALLOWED_BLURS",
+    "ALLOWED_TYPES",
+    "ALLOWED_CROPS",
+    "ALLOWED_PADDINGS",
+    "ALLOWED_INTERPOLATIONS",
+    "DTYPES_MAX",
+    "ALLOWED_COLOR_CONVERSIONS",
+    "ALLOWED_GRIDMASK_MODES",
+]

solt/constants/_constants.py

@@ -0,0 +1,17 @@
+import cv2
+import numpy as np
+
+ALLOWED_PADDINGS = {"z": cv2.BORDER_CONSTANT, "r": cv2.BORDER_REFLECT_101}
+ALLOWED_INTERPOLATIONS = {
+    "bilinear": cv2.INTER_LINEAR,
+    "bicubic": cv2.INTER_CUBIC,
+    "nearest": cv2.INTER_NEAREST,
+    "area": cv2.INTER_AREA,
+    "lanczos": cv2.INTER_LANCZOS4,
+}
+ALLOWED_CROPS = {"c", "r"}
+ALLOWED_TYPES = {"I", "M", "P", "L"}
+ALLOWED_BLURS = {"g", "m", "mo"}
+ALLOWED_COLOR_CONVERSIONS = {"gs2rgb", "rgb2gs", "none"}
+DTYPES_MAX = {np.dtype("uint8"): 255, np.dtype("uint16"): 65536}
+ALLOWED_GRIDMASK_MODES = {"crop", "reserve", "none"}

solt/core/__init__.py

@@ -0,0 +1,24 @@
+from ._core import Stream, SelectiveStream
+from ._base_transforms import (
+    BaseTransform,
+    MatrixTransform,
+)
+from ._base_transforms import (
+    PaddingPropertyHolder,
+    InterpolationPropertyHolder,
+    ImageTransform,
+)
+from ._data import DataContainer, Keypoints
+
+
+__all__ = [
+    "Stream",
+    "SelectiveStream",
+    "DataContainer",
+    "Keypoints",
+    "BaseTransform",
+    "MatrixTransform",
+    "PaddingPropertyHolder",
+    "InterpolationPropertyHolder",
+    "ImageTransform",
+]