Add cutmix_tf

pooya-mohammadi · Mar 6, 2024 · d7066e8 · d7066e8
1 parent c3c029e
commit d7066e8
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/deep_utils/__init__.py b/deep_utils/__init__.py
@@ -4,7 +4,7 @@
 from .utils.lib_utils.integeration_utils import import_lazy_module
 
 # Deep Utils version number
-__version__ = "1.3.22"
+__version__ = "1.3.23"
 
 from .utils.constants import DUMMY_PATH, Backends
 
@@ -70,6 +70,8 @@
 import_lazy_module("TorchAudioUtils", "audio.audio_utils.torchaudio_utils")
 import_lazy_module("UltralightTorchFaceDetector",
                    "vision.face_detection.ultralight.torch.ultralight_torch_face_detection")
+import_lazy_module("CutMixTF", "augmentation.cutmix.cutmix_tf")
+
 
 if TYPE_CHECKING:
     from .utils.box_utils.boxes import Box, Point
@@ -118,6 +120,7 @@
     from .medical.nib_utils.nib_utils import NIBUtils
     from .audio.audio_utils.torchaudio_utils import TorchAudioUtils
     from .medical.nnunet_utils.nnunet_utils import NNUnetUtils
+    from .augmentation.cutmix.cutmix_tf import CutMixTF
 else:
     import sys
 

diff --git a/deep_utils/dummy_objects/dummies.py b/deep_utils/dummy_objects/dummies.py
@@ -14,6 +14,10 @@ class TFUtils(metaclass=DummyObject):
     _backend = [Backends.TENSORFLOW]
 
 
+class CutMixTF(metaclass=DummyObject):
+    _backend = [Backends.TENSORFLOW]
+
+
 class TorchAudioUtils(metaclass=DummyObject):
     _backend = [Backends.TORCHAUDIO]
 

diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 import setuptools
 
-VERSION = "1.3.22"
+VERSION = "1.3.23"
 
 long_description = open("Readme.md", mode="r", encoding="utf-8").read()